Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
| 63 |
-
|
| 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 |
|