ermiaazarkhalili commited on
Commit
d935669
·
verified ·
1 Parent(s): 1a70f27

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +57 -1
README.md CHANGED
@@ -1,4 +1,60 @@
1
  ---
 
 
 
 
 
 
 
 
2
  datasets:
3
  - HuggingFaceM4/ChartQA
4
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ base_model: Qwen/Qwen2.5-VL-7B-Instruct
3
+ library_name: transformers
4
+ model_name: qwen2.5-7b-instruct-trl-sft-ChartQA
5
+ tags:
6
+ - generated_from_trainer
7
+ - trl
8
+ - sft
9
+ licence: license
10
  datasets:
11
  - HuggingFaceM4/ChartQA
12
+ ---
13
+
14
+ # Model Card for qwen2.5-7b-instruct-trl-sft-ChartQA
15
+
16
+ This model is a fine-tuned version of [Qwen/Qwen2.5-VL-7B-Instruct](https://huggingface.co/Qwen/Qwen2.5-VL-7B-Instruct).
17
+ It has been trained using [TRL](https://github.com/huggingface/trl).
18
+
19
+ ## Quick start
20
+
21
+ ```python
22
+ from transformers import pipeline
23
+
24
+ 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?"
25
+ generator = pipeline("text-generation", model="ermiaazarkhalili/qwen2.5-7b-instruct-trl-sft-ChartQA", device="cuda")
26
+ output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
27
+ print(output["generated_text"])
28
+ ```
29
+
30
+ ## Training procedure
31
+
32
+ [<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/behroozazarkhalili/qwen2.5-7b-instruct-trl-sft-ChartQA/runs/evvf8o2r)
33
+
34
+
35
+ This model was trained with SFT.
36
+
37
+ ### Framework versions
38
+
39
+ - TRL: 0.19.0
40
+ - Transformers: 4.53.0.dev0
41
+ - Pytorch: 2.5.1+cu121
42
+ - Datasets: 3.6.0
43
+ - Tokenizers: 0.21.1
44
+
45
+ ## Citations
46
+
47
+
48
+
49
+ Cite TRL as:
50
+
51
+ ```bibtex
52
+ @misc{vonwerra2022trl,
53
+ title = {{TRL: Transformer Reinforcement Learning}},
54
+ 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},
55
+ year = 2020,
56
+ journal = {GitHub repository},
57
+ publisher = {GitHub},
58
+ howpublished = {\url{https://github.com/huggingface/trl}}
59
+ }
60
+ ```