david-thrower commited on
Commit
ab1c97c
·
verified ·
1 Parent(s): b991544

Update app.py

Browse files

Update the params for generation ...

Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -16,7 +16,12 @@ from smolagents import (
16
 
17
  MODEL_NAME = "Qwen/Qwen3-0.6B"
18
 
19
- model = TransformersModel(model_id=MODEL_NAME)
 
 
 
 
 
20
 
21
  agent = CodeAgent(
22
  tools=[ DuckDuckGoSearchTool(),
 
16
 
17
  MODEL_NAME = "Qwen/Qwen3-0.6B"
18
 
19
+ model = TransformersModel(
20
+ model_id=MODEL_NAME,
21
+ max_new_tokens=4096,
22
+ temperature=0.6,
23
+ top_p=0.95,
24
+ repetition_penalty=1.1)
25
 
26
  agent = CodeAgent(
27
  tools=[ DuckDuckGoSearchTool(),