wt002 commited on
Commit
149648b
·
verified ·
1 Parent(s): a6bd097

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -56,7 +56,7 @@ class ZephyrToolCallingAgentTool(Tool):
56
  "text-generation",
57
  model=self.model,
58
  tokenizer=self.tokenizer,
59
- max_new_tokens=512,
60
  temperature=0.2
61
  )
62
 
 
56
  "text-generation",
57
  model=self.model,
58
  tokenizer=self.tokenizer,
59
+ max_new_tokens=128, # 128 or 512
60
  temperature=0.2
61
  )
62