End of training

README.md

@@ -1,13 +1,14 @@
 ---
 library_name: peft
-base_model: trl-internal-testing/tiny-random-LlamaForCausalLM
+license: apache-2.0
+base_model: openlm-research/open_llama_3b
 tags:
 - axolotl
 - generated_from_trainer
 datasets:
-- Aivesa/dataset_65ef80a6-755a-4c47-b107-9c0cb4da1f62
+- Aivesa/dataset_b98c9df0-5280-40e9-9125-37aafc6279e9
 model-index:
-- name: 1a20c85f-2d56-4d9b-a717-b6a1b3f4212d
+- name: cd10faed-0ece-4718-b1de-7fa9782b2746
   results: []
 ---
 
@@ -20,17 +21,18 @@ should probably proofread and complete it, then remove this comment. -->
 axolotl version: `0.6.0`
 ```yaml
 adapter: lora
-base_model: trl-internal-testing/tiny-random-LlamaForCausalLM
+base_model: openlm-research/open_llama_3b
 bf16: auto
 chat_template: llama3
 dataset_prepared_path: /workspace/axolotl/data/prepared
 datasets:
 - ds_type: json
   format: custom
-  path: Aivesa/dataset_65ef80a6-755a-4c47-b107-9c0cb4da1f62
+  path: Aivesa/dataset_b98c9df0-5280-40e9-9125-37aafc6279e9
   type:
-    field_instruction: title
-    field_output: content
+    field_input: sentence1
+    field_instruction: genre
+    field_output: sentence2
     system_format: '{system}'
     system_prompt: ''
 debug: null
@@ -46,7 +48,7 @@ fsdp_config: null
 gradient_accumulation_steps: 4
 gradient_checkpointing: false
 group_by_length: false
-hub_model_id: Aivesa/1a20c85f-2d56-4d9b-a717-b6a1b3f4212d
+hub_model_id: Aivesa/cd10faed-0ece-4718-b1de-7fa9782b2746
 hub_private_repo: true
 hub_repo: null
 hub_strategy: checkpoint
@@ -77,6 +79,8 @@ sample_packing: false
 save_safetensors: true
 saves_per_epoch: 4
 sequence_len: 512
+special_tokens:
+  pad_token: </s>
 strict: false
 tf32: false
 tokenizer_type: AutoTokenizer
@@ -86,10 +90,10 @@ use_accelerate: true
 val_set_size: 0.05
 wandb_entity: null
 wandb_mode: online
-wandb_name: 65ef80a6-755a-4c47-b107-9c0cb4da1f62
+wandb_name: b98c9df0-5280-40e9-9125-37aafc6279e9
 wandb_project: Gradients-On-Demand
 wandb_run: your_name
-wandb_runid: 65ef80a6-755a-4c47-b107-9c0cb4da1f62
+wandb_runid: b98c9df0-5280-40e9-9125-37aafc6279e9
 warmup_steps: 10
 weight_decay: 0.0
 xformers_attention: null
@@ -98,11 +102,11 @@ xformers_attention: null
 
 </details><br>
 
-# 1a20c85f-2d56-4d9b-a717-b6a1b3f4212d
+# cd10faed-0ece-4718-b1de-7fa9782b2746
 
-This model is a fine-tuned version of [trl-internal-testing/tiny-random-LlamaForCausalLM](https://huggingface.co/trl-internal-testing/tiny-random-LlamaForCausalLM) on the Aivesa/dataset_65ef80a6-755a-4c47-b107-9c0cb4da1f62 dataset.
+This model is a fine-tuned version of [openlm-research/open_llama_3b](https://huggingface.co/openlm-research/open_llama_3b) on the Aivesa/dataset_b98c9df0-5280-40e9-9125-37aafc6279e9 dataset.
 It achieves the following results on the evaluation set:
-- Loss: 10.3772
+- Loss: 0.9438
 
 ## Model description
 
@@ -136,9 +140,9 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch  | Step | Validation Loss |
 |:-------------:|:------:|:----:|:---------------:|
-| 10.377        | 0.0013 | 3    | 10.3776         |
-| 10.3812       | 0.0026 | 6    | 10.3774         |
-| 10.3764       | 0.0038 | 9    | 10.3772         |
+| 0.9494        | 0.0030 | 3    | 1.0563          |
+| 1.1032        | 0.0060 | 6    | 1.0397          |
+| 0.9986        | 0.0090 | 9    | 0.9438          |
 
 
 ### Framework versions