End of training

README.md

@@ -1,10 +1,11 @@
 ---
 license: apache-2.0
-base_model: PY007/TinyLlama-1.1B-Chat-v0.3
+library_name: peft
 tags:
 - trl
 - sft
 - generated_from_trainer
+base_model: PY007/TinyLlama-1.1B-Chat-v0.3
 model-index:
 - name: tinyllama-min-primary-secondary-lora
   results: []
@@ -42,7 +43,8 @@ The following hyperparameters were used during training:
 - total_train_batch_size: 32
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: cosine
-- training_steps: 200
+- num_epochs: 1
+- mixed_precision_training: Native AMP
 
 ### Training results
 
@@ -50,7 +52,8 @@ The following hyperparameters were used during training:
 
 ### Framework versions
 
-- Transformers 4.31.0
-- Pytorch 2.0.1+cu117
-- Datasets 2.15.0
-- Tokenizers 0.13.3
+- PEFT 0.10.0
+- Transformers 4.39.0
+- Pytorch 2.2.0+cu121
+- Datasets 2.18.0
+- Tokenizers 0.15.2