End of training

README.md

@@ -1,71 +1,50 @@
 ---
-license: mit
+license: apache-2.0
 base_model: TinyLlama/TinyLlama-1.1B-Chat-v1.0
 tags:
 - trl
 - sft
-- sgd
+- generated_from_trainer
 model-index:
 - name: mayo
   results: []
-datasets:
-- nroggendorff/mayo
-language:
-- en
 ---
 
-# Mayonnaise LLM
+<!-- This model card has been generated automatically according to the information the Trainer had access to. You
+should probably proofread and complete it, then remove this comment. -->
 
-Mayo is a language model fine-tuned on the [Mayo dataset](https://huggingface.co/datasets/nroggendorff/mayo) using Supervised Fine-Tuning (SFT) and Teacher Reinforced Learning (TRL) techniques. It is based on the [TinyLlama/TinyLlama-1.1B-Chat-v1.0 model](https://huggingface.co/TinyLlama/TinyLlama-1.1B-Chat-v1.0).
+# mayo
 
-## Features
+This model is a fine-tuned version of [TinyLlama/TinyLlama-1.1B-Chat-v1.0](https://huggingface.co/TinyLlama/TinyLlama-1.1B-Chat-v1.0) on an unknown dataset.
 
-- Utilizes SFT and TRL techniques for improved performance
-- Supports English language
+## Model description
 
-## Usage
+More information needed
 
-To use the Mayo LLM, you can load the model using the Hugging Face Transformers library:
+## Intended uses & limitations
 
-```python
-from transformers import pipeline
+More information needed
 
-pipe = pipeline("text-generation", model="nroggendorff/mayo")
+## Training and evaluation data
 
-question = "What color is the sky?"
-conv = [{"role": "user", "content": question}]
+More information needed
 
-response = pipe(conv, max_new_tokens=32)[0]['generated_text'][-1]['content']
-print(response)
-```
+## Training procedure
 
-To use the model with quantization:
+### Training hyperparameters
 
-```python
-from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
-import torch
+The following hyperparameters were used during training:
+- learning_rate: 0.0001
+- train_batch_size: 32
+- eval_batch_size: 16
+- seed: 42
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: linear
+- training_steps: 350
 
-bnb_config = BitsAndBytesConfig(
-    load_in_4bit=True,
-    bnb_4bit_use_double_quant=True,
-    bnb_4bit_quant_type="nf4",
-    bnb_4bit_compute_dtype=torch.bfloat16
-)
+### Framework versions
 
-model_id = "nroggendorff/mayo"
-
-tokenizer = AutoTokenizer.from_pretrained(model_id)
-model = AutoModelForCausalLM.from_pretrained(model_id, quantization_config=bnb_config)
-
-prompt = "<|user|>\nWhat color is the sky?</s>\n"
-inputs = tokenizer(prompt, return_tensors="pt")
-
-outputs = model.generate(**inputs, max_new_tokens=32)
-
-generated_text = tokenizer.batch_decode(outputs)[0]
-print(generated_text)
-```
-
-## License
-
-This project is licensed under the MIT License.
+- Transformers 4.39.3
+- Pytorch 2.1.2
+- Datasets 2.18.0
+- Tokenizers 0.15.2

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:98550081dc345f70f3b835d7bd4a709d55de1bafaed609303c1ab6a5d926ab5b
+oid sha256:f12a3986dca4a2d8406fdb25af0f46a9a83b2e7b3daab798068571c736d36175
 size 4400216536

runs/Jun03_21-02-51_c363bad361ad/events.out.tfevents.1717448584.c363bad361ad.34.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:93b4a2f10f0e86121bda5b88ca5e2ea7bd0f60a1e9b6a000d11763025cf596b5
+size 8736

tokenizer.json

@@ -1,6 +1,11 @@
 {
   "version": "1.0",
-  "truncation": null,
+  "truncation": {
+    "direction": "Right",
+    "max_length": 512,
+    "strategy": "LongestFirst",
+    "stride": 0
+  },
   "padding": null,
   "added_tokens": [
     {

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e8994d669d7f58dd765f8f16f3850c8523c83adc26e951c258408f22c5ef0d68
+oid sha256:9a1f2e088c534535543f338026ae118e0567c0b852073737ae6b359bcbdc2513
 size 4920