Spaces:
Runtime error
Runtime error
Mikhil-jivus
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -49,7 +49,7 @@ def respond(
|
|
49 |
# Generate a response
|
50 |
chat_history_ids = model.generate(
|
51 |
input_ids,
|
52 |
-
|
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,
|