Spaces:
Runtime error
Runtime error
Commit
·
45883da
1
Parent(s):
6f69561
Update app.py
Browse files
app.py
CHANGED
@@ -158,6 +158,9 @@ def generate(
|
|
158 |
attention_mask=attention_mask,
|
159 |
max_new_tokens=30,
|
160 |
num_beams=3,
|
|
|
|
|
|
|
161 |
)
|
162 |
|
163 |
gen_text = tokenizer.decode(
|
|
|
158 |
attention_mask=attention_mask,
|
159 |
max_new_tokens=30,
|
160 |
num_beams=3,
|
161 |
+
do_sample=True,
|
162 |
+
temperature=0.7,
|
163 |
+
top_k=0,
|
164 |
)
|
165 |
|
166 |
gen_text = tokenizer.decode(
|