gregH commited on
Commit
40686c5
1 Parent(s): 7e11532

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -110,7 +110,7 @@ def engine(input_embeds):
110
  batch_input_embeds = input_embeds[start:start+batch_size]
111
  outputs = m.generate(
112
  inputs_embeds = batch_input_embeds.to("cuda"),
113
- max_new_tokens = 500,
114
  do_sample = True,
115
  temperature = 0.6,
116
  top_p = 0.9,
 
110
  batch_input_embeds = input_embeds[start:start+batch_size]
111
  outputs = m.generate(
112
  inputs_embeds = batch_input_embeds.to("cuda"),
113
+ max_new_tokens = 512,
114
  do_sample = True,
115
  temperature = 0.6,
116
  top_p = 0.9,