ysharma HF staff commited on
Commit
f44feff
1 Parent(s): cb25f52

updates to return tokens

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ def text_generate(problem, template, prompt):
37
  {
38
  "top_p": 0.9,
39
  "temperature": 1.1,
40
- "max_new_tokens": 250,
41
  "return_full_text": True
42
  }, "options":
43
  {
 
37
  {
38
  "top_p": 0.9,
39
  "temperature": 1.1,
40
+ "max_new_tokens": 64,
41
  "return_full_text": True
42
  }, "options":
43
  {