Mikhil-jivus commited on
Commit
ae4333a
·
verified ·
1 Parent(s): b7d6aa3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ def respond(
49
  # Generate a response
50
  chat_history_ids = model.generate(
51
  input_ids,
52
- max_length=max_tokens,
53
  temperature=temperature,
54
  top_p=top_p,
55
  pad_token_id=tokenizer.eos_token_id,
 
49
  # Generate a response
50
  chat_history_ids = model.generate(
51
  input_ids,
52
+ max_new_tokens=max_tokens,
53
  temperature=temperature,
54
  top_p=top_p,
55
  pad_token_id=tokenizer.eos_token_id,