matsuo-lab
commited on
Commit
•
f72f467
1
Parent(s):
3f76763
Update README.md
Browse files
README.md
CHANGED
@@ -72,10 +72,7 @@ with torch.no_grad():
|
|
72 |
max_new_tokens=100,
|
73 |
do_sample=True,
|
74 |
temperature=0.7,
|
75 |
-
top_p=0.95
|
76 |
-
pad_token_id=tokenizer.pad_token_id,
|
77 |
-
bos_token_id=tokenizer.bos_token_id,
|
78 |
-
eos_token_id=tokenizer.eos_token_id
|
79 |
)
|
80 |
|
81 |
output = tokenizer.decode(output_ids.tolist()[0])
|
|
|
72 |
max_new_tokens=100,
|
73 |
do_sample=True,
|
74 |
temperature=0.7,
|
75 |
+
top_p=0.95
|
|
|
|
|
|
|
76 |
)
|
77 |
|
78 |
output = tokenizer.decode(output_ids.tolist()[0])
|