Model Card for Lite-Oute-1-65M-smol-smoltalk
This model is a fine-tuned version of OuteAI/Lite-Oute-1-65M using smol-smoltalk dataset for 1 epoch. Lite-Oute-1-65M (Base) is an experimental ultra-compact base model in the Lite series, built on the LLaMA architecture and comprising approximately 65 million parameters. It has been trained using TRL. Below the finetuned model is evaluated on common benchmarks.
Benchmarks:
Metric | Lite-Oute-1-65M | Lite-Oute-1-65M-smol-smoltalk | SmolLM2-135M-Instruct |
---|---|---|---|
HellaSwag | 28.27 | 28.56 | 42.83 |
ARC (Average) | 31.85 | 28.275 | 38.37 |
PIQA | 60.45 | 59.47 | 66.7 |
Winogrande | 51.7 | 53.59 | 51.54 |
MMLU | 23.05 | 22.9 | 22.96 |
Quick start
from transformers import pipeline
question = "If you had a time machine, but could only go to the past or the future once and never return, which would you choose and why?"
generator = pipeline("text-generation", model="Ellight/Lite-Oute-1-65M-smol-smoltalk", device="cuda")
output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
print(output["generated_text"])
Training procedure
This model was trained with SFT.
Risk Disclaimer
By using this model, you acknowledge that you understand and assume the risks associated with its use. You are solely responsible for ensuring compliance with all applicable laws and regulations. We disclaim any liability for problems arising from the use of this open-source model, including but not limited to direct, indirect, incidental, consequential, or punitive damages. We make no warranties, express or implied, regarding the model's performance, accuracy, or fitness for a particular purpose. Your use of this model is at your own risk, and you agree to hold harmless and indemnify us, our affiliates, and our contributors from any claims, damages, or expenses arising from your use of the model.
Framework versions
- TRL: 0.12.1
- Transformers: 4.46.3
- Pytorch: 2.2.1+cu121
- Datasets: 3.1.0
- Tokenizers: 0.20.3
Citations
Cite TRL as:
@misc{vonwerra2022trl,
title = {{TRL: Transformer Reinforcement Learning}},
author = {Leandro von Werra and Younes Belkada and Lewis Tunstall and Edward Beeching and Tristan Thrush and Nathan Lambert and Shengyi Huang and Kashif Rasul and Quentin Gallouédec},
year = 2020,
journal = {GitHub repository},
publisher = {GitHub},
howpublished = {\url{https://github.com/huggingface/trl}}
}
- Downloads last month
- 59