fuvty commited on
Commit
de2ba9d
·
1 Parent(s): 5ccf219

[debug] run on CPU

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = "cuda"
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.