File size: 1,589 Bytes
de9221d
 
 
 
 
 
9ddd17f
de9221d
 
9ddd17f
de9221d
 
a4264be
de9221d
a4264be
de9221d
 
a4264be
de9221d
a4264be
 
de9221d
a4264be
de9221d
 
 
 
a4264be
 
de9221d
a4264be
de9221d
a4264be
 
 
de9221d
a4264be
de9221d
a4264be
de9221d
 
 
 
 
a4264be
de9221d
a4264be
 
 
de9221d
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
---
base_model: Qwen/Qwen2.5-Coder-0.5B-Instruct
library_name: transformers
model_name: tiny-coder-prompt-completion-0.5B
tags:
- generated_from_trainer
- trackio:https://huggingface.co/spaces/finnvoorhees/huggingface-static-ab58c9
- trl
- sft
- trackio
licence: license
---

# Model Card for tiny-coder-prompt-completion-0.5B

This model is a fine-tuned version of [Qwen/Qwen2.5-Coder-0.5B-Instruct](https://huggingface.co/Qwen/Qwen2.5-Coder-0.5B-Instruct).
It has been trained using [TRL](https://github.com/huggingface/trl).

## Quick start

```python
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="finnvoorhees/tiny-coder-prompt-completion-0.5B", 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.

### Framework versions

- TRL: 1.2.0
- Transformers: 5.6.2
- Pytorch: 2.11.0
- Datasets: 4.8.4
- Tokenizers: 0.22.2

## Citations



Cite TRL as:
    
```bibtex
@software{vonwerra2020trl,
  title   = {{TRL: Transformers Reinforcement Learning}},
  author  = {von Werra, Leandro and Belkada, Younes and Tunstall, Lewis and Beeching, Edward and Thrush, Tristan and Lambert, Nathan and Huang, Shengyi and Rasul, Kashif and Gallouédec, Quentin},
  license = {Apache-2.0},
  url     = {https://github.com/huggingface/trl},
  year    = {2020}
}
```