mimimumumomo commited on
Commit
fc905e8
·
verified ·
1 Parent(s): 57c97a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -57,13 +57,15 @@ final_answer = FinalAnswerTool()
57
  # temperature=0.7,
58
  # )
59
 
 
 
60
  model = TransformersModel(
61
  model_id="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
62
- dtype="auto",
63
- max_tokens=1024,
64
- temperature=0.5
65
  )
66
 
 
67
  # Import tool from Hub
68
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
69
 
 
57
  # temperature=0.7,
58
  # )
59
 
60
+
61
+
62
  model = TransformersModel(
63
  model_id="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
64
+ max_new_tokens=512, # TinyLlama 한계 맞춤
65
+ temperature=0.7, # optional
 
66
  )
67
 
68
+
69
  # Import tool from Hub
70
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
71