ford442 commited on
Commit
b05ba5b
·
verified ·
1 Parent(s): 89150fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -126,7 +126,7 @@ def infer(
126
  outputs = model.generate(
127
  input_ids=input_ids,
128
  attention_mask=attention_mask,
129
- max_new_tokens=77,
130
  temperature=0.2,
131
  top_p=0.9,
132
  do_sample=True,
 
126
  outputs = model.generate(
127
  input_ids=input_ids,
128
  attention_mask=attention_mask,
129
+ max_new_tokens=65,
130
  temperature=0.2,
131
  top_p=0.9,
132
  do_sample=True,