AmirMohseni commited on
Commit
75981ee
·
verified ·
1 Parent(s): 75c3533

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -1,7 +1,7 @@
1
  ---
2
  base_model: google/gemma-2-9b-it
3
  library_name: transformers
4
- model_name: Gemma-9b-WithHint-v1
5
  tags:
6
  - generated_from_trainer
7
  - trl
@@ -20,7 +20,7 @@ It has been trained using [TRL](https://github.com/huggingface/trl).
20
  from transformers import pipeline
21
 
22
  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?"
23
- generator = pipeline("text-generation", model="AmirMohseni/Gemma-9b-WithHint-v1", device="cuda")
24
  output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
25
  print(output["generated_text"])
26
  ```
 
1
  ---
2
  base_model: google/gemma-2-9b-it
3
  library_name: transformers
4
+ model_name: Velma-9b
5
  tags:
6
  - generated_from_trainer
7
  - trl
 
20
  from transformers import pipeline
21
 
22
  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?"
23
+ generator = pipeline("text-generation", model="AmirMohseni/Velma-9b", device="cuda")
24
  output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
25
  print(output["generated_text"])
26
  ```