limhyeonseok
commited on
Commit
•
f361a34
1
Parent(s):
36fdb96
Update README.md
Browse files
README.md
CHANGED
@@ -180,8 +180,7 @@ outputs = model.generate(
|
|
180 |
eos_token_id=terminators,
|
181 |
do_sample=True,
|
182 |
temperature=0.6,
|
183 |
-
top_p=0.9
|
184 |
-
repetition_penalty = 1.1
|
185 |
)
|
186 |
|
187 |
print(tokenizer.decode(outputs[0][input_ids.shape[-1]:], skip_special_tokens=True))
|
|
|
180 |
eos_token_id=terminators,
|
181 |
do_sample=True,
|
182 |
temperature=0.6,
|
183 |
+
top_p=0.9
|
|
|
184 |
)
|
185 |
|
186 |
print(tokenizer.decode(outputs[0][input_ids.shape[-1]:], skip_special_tokens=True))
|