blankenshipdanielj/gitgraph-commit-generator

1---2base_model: Qwen/Qwen2.5-1.5B-Instruct3library_name: transformers4model_name: gitgraph-commit-generator5tags:6- generated_from_trainer7- hf_jobs8- trl9- sft10licence: license11---12 13# Model Card for gitgraph-commit-generator14 15This model is a fine-tuned version of [Qwen/Qwen2.5-1.5B-Instruct](https://huggingface.co/Qwen/Qwen2.5-1.5B-Instruct).16It has been trained using [TRL](https://github.com/huggingface/trl).17 18## Quick start19 20```python21from transformers import pipeline22 23question = "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?"24generator = pipeline("text-generation", model="blankenshipdanielj/gitgraph-commit-generator", device="cuda")25output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]26print(output["generated_text"])27```28 29## Training procedure30 31 32 33 34This model was trained with SFT.35 36### Framework versions37 38- TRL: 0.27.139- Transformers: 5.0.040- Pytorch: 2.10.041- Datasets: 4.5.042- Tokenizers: 0.22.243 44## Citations45 46 47 48Cite TRL as:49    50```bibtex51@misc{vonwerra2022trl,52	title        = {{TRL: Transformer Reinforcement Learning}},53	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{\'e}dec},54	year         = 2020,55	journal      = {GitHub repository},56	publisher    = {GitHub},57	howpublished = {\url{https://github.com/huggingface/trl}}58}59```