Spaces:
Sleeping
Sleeping
[debug] run on CPU
Browse files
app.py
CHANGED
|
@@ -34,7 +34,7 @@ class ModelManager:
|
|
| 34 |
t2t_context_model: str = "Qwen/Qwen3-0.6B",
|
| 35 |
t2t_answer_model: str = "Qwen/Qwen2.5-0.5B-Instruct",
|
| 36 |
c2c_checkpoint_path: str = "local/checkpoints/qwen3_0.6b+qwen2.5_0.5b_Fuser",
|
| 37 |
-
device: str = "
|
| 38 |
):
|
| 39 |
"""
|
| 40 |
Initialize ModelManager with model configurations.
|
|
|
|
| 34 |
t2t_context_model: str = "Qwen/Qwen3-0.6B",
|
| 35 |
t2t_answer_model: str = "Qwen/Qwen2.5-0.5B-Instruct",
|
| 36 |
c2c_checkpoint_path: str = "local/checkpoints/qwen3_0.6b+qwen2.5_0.5b_Fuser",
|
| 37 |
+
device: str = "auto"
|
| 38 |
):
|
| 39 |
"""
|
| 40 |
Initialize ModelManager with model configurations.
|