Update app.py
Browse files
app.py
CHANGED
@@ -256,7 +256,7 @@ def generate(
|
|
256 |
qa_prompt,
|
257 |
system_prompt=system_prompt,
|
258 |
#stream=True,
|
259 |
-
|
260 |
temperature=temperature,
|
261 |
top_k=top_k,
|
262 |
top_p=top_p,
|
|
|
256 |
qa_prompt,
|
257 |
system_prompt=system_prompt,
|
258 |
#stream=True,
|
259 |
+
max_tokens=max_new_tokens,
|
260 |
temperature=temperature,
|
261 |
top_k=top_k,
|
262 |
top_p=top_p,
|