Commit
•
2dc7fa1
1
Parent(s):
e34c599
Update README.md
Browse files
README.md
CHANGED
|
@@ -71,8 +71,9 @@ def gen(lan="en", x=""):
|
|
| 71 |
return_token_type_ids=False
|
| 72 |
).to("cuda"),
|
| 73 |
max_new_tokens=2000,
|
| 74 |
-
temperature=0.
|
| 75 |
-
|
|
|
|
| 76 |
stopping_criteria=stopping_criteria
|
| 77 |
)
|
| 78 |
return tokenizer.decode(gened[0][1:]).replace(prompt+" ", "").replace("</끝>", "")
|
|
|
|
| 71 |
return_token_type_ids=False
|
| 72 |
).to("cuda"),
|
| 73 |
max_new_tokens=2000,
|
| 74 |
+
temperature=0.3,
|
| 75 |
+
# no_repeat_ngram_size=5,
|
| 76 |
+
num_beams=5,
|
| 77 |
stopping_criteria=stopping_criteria
|
| 78 |
)
|
| 79 |
return tokenizer.decode(gened[0][1:]).replace(prompt+" ", "").replace("</끝>", "")
|