Tao Wu commited on
Commit
a25d322
1 Parent(s): b2dd7b6

update exp prompt

Browse files
Files changed (1) hide show
  1. app/embedding_setup.py +1 -1
app/embedding_setup.py CHANGED
@@ -171,7 +171,7 @@ def generate_exp(
171
  top_p=1.0,
172
  top_k=40,
173
  num_beams=1,
174
- max_new_tokens=256,
175
  batch_size=1,
176
  **kwargs,
177
  ):
 
171
  top_p=1.0,
172
  top_k=40,
173
  num_beams=1,
174
+ max_new_tokens=512,
175
  batch_size=1,
176
  **kwargs,
177
  ):