vpcom commited on
Commit
18b506f
·
1 Parent(s): a3aed2a

fix: use less future tokens

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -147,7 +147,7 @@ def evolve_text(x):
147
  possible_generation = "".join([x for x in generate(
148
  x, "", additional_inputs[0].value,
149
  temperature=additional_inputs[1].value,
150
- max_new_tokens=additional_inputs[2].value,
151
  top_p=additional_inputs[3].value,
152
  repetition_penalty=additional_inputs[4].value,
153
  seed=42,
 
147
  possible_generation = "".join([x for x in generate(
148
  x, "", additional_inputs[0].value,
149
  temperature=additional_inputs[1].value,
150
+ max_new_tokens=10,
151
  top_p=additional_inputs[3].value,
152
  repetition_penalty=additional_inputs[4].value,
153
  seed=42,