gregH commited on
Commit
e4c096c
1 Parent(s): 5587284

Update app.py

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