Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -32,7 +32,7 @@ def predict(message, history):
|
|
32 |
top_k=5,
|
33 |
temperature=0.5,
|
34 |
repetition_penalty=1.2,
|
35 |
-
max_new_tokens=
|
36 |
eos_token_id=tokenizer.eos_token_id,
|
37 |
pad_token_id=tokenizer.eos_token_id
|
38 |
)
|
|
|
32 |
top_k=5,
|
33 |
temperature=0.5,
|
34 |
repetition_penalty=1.2,
|
35 |
+
max_new_tokens=64,
|
36 |
eos_token_id=tokenizer.eos_token_id,
|
37 |
pad_token_id=tokenizer.eos_token_id
|
38 |
)
|