Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -69,6 +69,7 @@ def chat(message, history, system, temperature, do_sample, max_tokens):
|
|
69 |
"do_sample": do_sample,
|
70 |
"temperature": temperature,
|
71 |
"eos_token_id": terminators,
|
|
|
72 |
}
|
73 |
|
74 |
thread = Thread(target=model.generate, kwargs=generate_kwargs)
|
|
|
69 |
"do_sample": do_sample,
|
70 |
"temperature": temperature,
|
71 |
"eos_token_id": terminators,
|
72 |
+
"pad_token_id": tokenizer.eos_token_id
|
73 |
}
|
74 |
|
75 |
thread = Thread(target=model.generate, kwargs=generate_kwargs)
|