Alphatao commited on
Commit
2a2e2c6
·
verified ·
1 Parent(s): 618ec19

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +0 -54
README.md CHANGED
@@ -1,57 +1,3 @@
1
  ---
2
- library_name: transformers
3
- model_name: Qwen-sft
4
- tags:
5
- - generated_from_trainer
6
- - trl
7
- - sft
8
- licence: license
9
- ---
10
-
11
- # Model Card for Qwen-sft
12
-
13
- This model is a fine-tuned version of [None](https://huggingface.co/None).
14
- It has been trained using [TRL](https://github.com/huggingface/trl).
15
-
16
- ## Quick start
17
-
18
- ```python
19
- from transformers import pipeline
20
-
21
- 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?"
22
- generator = pipeline("text-generation", model="Alphatao/Qwen-sft", device="cuda")
23
- output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
24
- print(output["generated_text"])
25
- ```
26
-
27
- ## Training procedure
28
-
29
- [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="150" height="24"/>](https://wandb.ai/alphatao-alphatao/Gradients-On-Demand/runs/k8u8gv12)
30
-
31
-
32
- This model was trained with SFT.
33
-
34
- ### Framework versions
35
-
36
- - TRL: 0.20.0
37
- - Transformers: 4.54.1
38
- - Pytorch: 2.7.1
39
- - Datasets: 4.0.0
40
- - Tokenizers: 0.21.4
41
-
42
- ## Citations
43
-
44
-
45
 
46
- Cite TRL as:
47
-
48
- ```bibtex
49
- @misc{vonwerra2022trl,
50
- title = {{TRL: Transformer Reinforcement Learning}},
51
- 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},
52
- year = 2020,
53
- journal = {GitHub repository},
54
- publisher = {GitHub},
55
- howpublished = {\url{https://github.com/huggingface/trl}}
56
- }
57
  ```
 
1
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
 
 
 
 
 
 
 
 
 
 
 
 
3
  ```