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: 16
+- eval_batch_size: 16
+- seed: 42
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: linear
+- training_steps: 650
 
-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:4678b8c283713bb83d9fe047db16f087707bc5b2c53206fe90af9c0c6f4878cd
+oid sha256:98550081dc345f70f3b835d7bd4a709d55de1bafaed609303c1ab6a5d926ab5b
 size 4400216536

runs/Jun03_20-42-00_a7f4754c6b47/events.out.tfevents.1717447323.a7f4754c6b47.34.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:49e7e563abc07d36027bf66b7005cfd7257a7d5f44287791b3de8c911381e400
+size 9192

runs/Jun03_20-46-02_a7f4754c6b47/events.out.tfevents.1717447572.a7f4754c6b47.34.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6f238cafa2faf83b40b86fe91f856b602c4cbb315a529273575128d90af2a907
+size 4640

runs/Jun03_20-47-10_a7f4754c6b47/events.out.tfevents.1717447634.a7f4754c6b47.34.2

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2f7ee633ce478b8cdf313b69048a2b2a487ae61acb99dad4c9cf12f8a5aee023
+size 4184

tokenizer.json

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

training_args.bin

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