Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -200,6 +200,9 @@ class Qwen3VLAutoThinkDemo:
|
|
| 200 |
Returns:
|
| 201 |
Dictionary containing response and metadata
|
| 202 |
"""
|
|
|
|
|
|
|
|
|
|
| 203 |
# Prepare message
|
| 204 |
message = [{"role": "system", "content": self.system_prompt}]
|
| 205 |
content_parts = []
|
|
|
|
| 200 |
Returns:
|
| 201 |
Dictionary containing response and metadata
|
| 202 |
"""
|
| 203 |
+
if self.model.device.type != "cuda":
|
| 204 |
+
self.model.to("cuda")
|
| 205 |
+
|
| 206 |
# Prepare message
|
| 207 |
message = [{"role": "system", "content": self.system_prompt}]
|
| 208 |
content_parts = []
|