efederici commited on
Commit
bcbabd5
1 Parent(s): 97a7b64

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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)