matsuo-lab
commited on
Commit
•
865ea56
1
Parent(s):
59e1284
Update README.md
Browse files
README.md
CHANGED
@@ -83,10 +83,7 @@ with torch.no_grad():
|
|
83 |
max_new_tokens=100,
|
84 |
do_sample=True,
|
85 |
temperature=0.7,
|
86 |
-
top_p=0.95
|
87 |
-
pad_token_id=tokenizer.pad_token_id,
|
88 |
-
bos_token_id=tokenizer.bos_token_id,
|
89 |
-
eos_token_id=tokenizer.eos_token_id
|
90 |
)
|
91 |
|
92 |
output = tokenizer.decode(output_ids.tolist()[0])
|
|
|
83 |
max_new_tokens=100,
|
84 |
do_sample=True,
|
85 |
temperature=0.7,
|
86 |
+
top_p=0.95
|
|
|
|
|
|
|
87 |
)
|
88 |
|
89 |
output = tokenizer.decode(output_ids.tolist()[0])
|