MultiTrickFox commited on
Commit
6822221
1 Parent(s): a80e3ea

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -4
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, do_sample=True,
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,