MultiTrickFox
commited on
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,
|