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