Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -21,7 +21,8 @@ def chat(prompt):
|
|
21 |
max_length=1024,
|
22 |
num_return_sequences=1,
|
23 |
top_p=0.9,
|
24 |
-
top_k=
|
|
|
25 |
pad_token_id=model.config.eos_token_id
|
26 |
)
|
27 |
|
|
|
21 |
max_length=1024,
|
22 |
num_return_sequences=1,
|
23 |
top_p=0.9,
|
24 |
+
top_k=50,
|
25 |
+
num_beams=2,
|
26 |
pad_token_id=model.config.eos_token_id
|
27 |
)
|
28 |
|