Qwen3-TTS

Runtime error

Sovenok-Hacker commited on Jan 25

Commit

2de7e08

verified ·

1 Parent(s): c867542

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -35,7 +35,7 @@ def get_model(model_type: str, model_size: str):
         model_path = get_model_path(model_type, model_size)
         loaded_models[key] = Qwen3TTSModel.from_pretrained(
             model_path,
-            device_map="cuda",
             dtype=torch.bfloat16,
             token=HF_TOKEN,
 #           attn_implementation="flash_attention_2",

         model_path = get_model_path(model_type, model_size)
         loaded_models[key] = Qwen3TTSModel.from_pretrained(
             model_path,
+            device_map="cpu",
             dtype=torch.bfloat16,
             token=HF_TOKEN,
 #           attn_implementation="flash_attention_2",