komus commited on
Commit
4f1c6b6
·
verified ·
1 Parent(s): 7f31da5

Update README.md

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