File size: 1,429 Bytes
1ad7c25
13933be
1ad7c25
4342b67
13933be
 
 
4342b67
13933be
1ad7c25
 
4342b67
1ad7c25
13933be
 
1ad7c25
13933be
1ad7c25
13933be
 
1ad7c25
13933be
4342b67
13933be
 
 
1ad7c25
13933be
1ad7c25
13933be
1ad7c25
 
 
13933be
1ad7c25
13933be
1ad7c25
13933be
4342b67
13933be
 
 
1ad7c25
13933be
1ad7c25
 
 
13933be
 
 
 
 
 
 
 
 
 
 
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
---
base_model: Qwen/Qwen2.5-1.5B-Instruct
library_name: transformers
model_name: model_sft_lora
tags:
- generated_from_trainer
- trl
- sft
licence: license
---

# Model Card for model_sft_lora

This model is a fine-tuned version of [Qwen/Qwen2.5-1.5B-Instruct](https://huggingface.co/Qwen/Qwen2.5-1.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="allout2726/model_sft_lora", 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.0.0
- Transformers: 5.5.0
- Pytorch: 2.10.0+cu128
- 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}
}
```