davidizzle commited on
Commit
4df0539
·
1 Parent(s): ab74be1

more tokens

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -89,7 +89,7 @@ if st.button("Generate"):
89
  with torch.no_grad():
90
  outputs = model.generate( **inputs,
91
  # max_new_tokens=100,
92
- max_new_tokens=256,
93
  do_sample=True,
94
  temperature=0.6,
95
  top_p=0.95,
 
89
  with torch.no_grad():
90
  outputs = model.generate( **inputs,
91
  # max_new_tokens=100,
92
+ max_new_tokens=512,
93
  do_sample=True,
94
  temperature=0.6,
95
  top_p=0.95,