JDhruv14 commited on
Commit
f1dffd7
·
verified ·
1 Parent(s): 0647748

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -81,7 +81,8 @@ def infer_text(history, system_text=""):
81
  system_text=system_text,
82
  temperature=0.7,
83
  top_p=0.9,
84
- max_new=256,
 
85
  )
86
 
87
  gen_cfg = GenerationConfig(**gen_cfg_kwargs)
 
81
  system_text=system_text,
82
  temperature=0.7,
83
  top_p=0.9,
84
+ max_new=512,
85
+ max_new=128,
86
  )
87
 
88
  gen_cfg = GenerationConfig(**gen_cfg_kwargs)