Commit
·
6822221
1
Parent(s):
a80e3ea
Update README.md
Browse files
README.md
CHANGED
|
@@ -18,15 +18,14 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
| 18 |
tokenizer = AutoTokenizer.from_pretrained("MultiTrickFox/bloom-2b5_Zen")
|
| 19 |
model = AutoModelForCausalLM.from_pretrained("MultiTrickFox/bloom-2b5_Zen")
|
| 20 |
|
| 21 |
-
tokenizer.pad_token_id = tokenizer.eos_token_id
|
| 22 |
-
|
| 23 |
generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
|
| 24 |
|
| 25 |
inp = [ """Today""", """Yesterday""" ]
|
| 26 |
|
| 27 |
out = generator(
|
| 28 |
-
inp,
|
| 29 |
-
|
|
|
|
| 30 |
temperature=.7,
|
| 31 |
typical_p=.6,
|
| 32 |
#top_p=.9,
|
|
|
|
| 18 |
tokenizer = AutoTokenizer.from_pretrained("MultiTrickFox/bloom-2b5_Zen")
|
| 19 |
model = AutoModelForCausalLM.from_pretrained("MultiTrickFox/bloom-2b5_Zen")
|
| 20 |
|
|
|
|
|
|
|
| 21 |
generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
|
| 22 |
|
| 23 |
inp = [ """Today""", """Yesterday""" ]
|
| 24 |
|
| 25 |
out = generator(
|
| 26 |
+
inp,
|
| 27 |
+
|
| 28 |
+
do_sample=True,
|
| 29 |
temperature=.7,
|
| 30 |
typical_p=.6,
|
| 31 |
#top_p=.9,
|