1inkusFace commited on
Commit
6886b34
·
verified ·
1 Parent(s): dd3b213

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -344,7 +344,7 @@ def expand_prompt(prompt):
344
  outputs = model.generate(
345
  input_ids=input_ids,
346
  attention_mask=attention_mask,
347
- max_new_tokens=384,
348
  temperature=0.2,
349
  top_p=0.9,
350
  do_sample=True,
 
344
  outputs = model.generate(
345
  input_ids=input_ids,
346
  attention_mask=attention_mask,
347
+ max_new_tokens=1024,
348
  temperature=0.2,
349
  top_p=0.9,
350
  do_sample=True,