Training complete

README.md

@@ -1,11 +1,12 @@
 ---
+library_name: transformers
+license: llama2
 base_model: meta-llama/Llama-2-7b-hf
 tags:
 - trl
 - sft
-- missing lyric Llama2
-- generated_from_trainer
 - missing lyric Llama2 1
+- generated_from_trainer
 datasets:
 - generator
 model-index:
@@ -46,7 +47,7 @@ The following hyperparameters were used during training:
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: constant_with_warmup
 - lr_scheduler_warmup_ratio: 0.04
-- num_epochs: 3
+- num_epochs: 1
 - mixed_precision_training: Native AMP
 
 ### Training results
@@ -55,7 +56,7 @@ The following hyperparameters were used during training:
 
 ### Framework versions
 
-- Transformers 4.39.3
-- Pytorch 2.1.2
+- Transformers 4.44.2
+- Pytorch 2.2.2+cu121
 - Datasets 2.18.0
-- Tokenizers 0.15.2
+- Tokenizers 0.19.1

generation_config.json

@@ -6,5 +6,5 @@
   "pad_token_id": 0,
   "temperature": 0.6,
   "top_p": 0.9,
-  "transformers_version": "4.39.3"
+  "transformers_version": "4.44.2"
 }