Update README.md

README.md

@@ -8,6 +8,8 @@ model-index:
   results: []
 library_name: peft
 inference: false
+datasets:
+- linkanjarad/baize-chat-data
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -15,7 +17,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # PythiaChat-2.8B_v0.1
 
-This model is a fine-tuned version of [EleutherAI/pythia-2.8b-deduped](https://huggingface.co/EleutherAI/pythia-2.8b-deduped) on the [Baize dataset](https://huggingface.co/datasets/linkanjarad/baize-chat-data/viewer/linkanjarad--baize-chat-data), trained for only 200+ steps for testing. This model is trained for "chat" style instruction following capabilities.
+This model is a fine-tuned version of [EleutherAI/pythia-2.8b-deduped](https://huggingface.co/EleutherAI/pythia-2.8b-deduped) on the [Baize dataset](https://huggingface.co/datasets/linkanjarad/baize-chat-data/viewer/linkanjarad--baize-chat-data) with LoRA, trained for only 200+ steps for testing. This model is trained for "chat" style instruction following capabilities.
 
 # Sample Use
 
@@ -85,4 +87,4 @@ The following hyperparameters were used during training:
 - Transformers 4.31.0
 - Pytorch 2.0.0
 - Datasets 2.13.1
-- Tokenizers 0.13.3
+- Tokenizers 0.13.3