Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -120,6 +120,7 @@ def evaluate(
|
|
120 |
top_p=top_p,
|
121 |
top_k=top_k,
|
122 |
num_beams=num_beams,
|
|
|
123 |
**kwargs,
|
124 |
)
|
125 |
with torch.no_grad():
|
|
|
120 |
top_p=top_p,
|
121 |
top_k=top_k,
|
122 |
num_beams=num_beams,
|
123 |
+
do_sample=True,
|
124 |
**kwargs,
|
125 |
)
|
126 |
with torch.no_grad():
|