groupuser/phi-1_5
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update README.md

Browse Source
This commit is contained in:
Gunasekar 2023-09-11 07:35:26 +00:00 committed by huggingface-web
parent cd4510ca85
commit 6ddac37bb9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -101,7 +101,7 @@ def print_prime(n):
Print all primes between 1 and n
"""''', return_tensors="pt", return_attention_mask=False)
eos_token_id = tokenizer.encode("```")[0]
eos_token_id = tokenizer.encode("``<|endoftext|>") # generation ends at `` or <|endoftext|>
outputs = model.generate(**inputs, max_length=500)
text = tokenizer.batch_decode(outputs)[0]
print(text)