Update app.py
Browse files
app.py
CHANGED
|
@@ -36,7 +36,7 @@ def get_model(model_type: str, model_size: str):
|
|
| 36 |
loaded_models[key] = Qwen3TTSModel.from_pretrained(
|
| 37 |
model_path,
|
| 38 |
device_map="cpu",
|
| 39 |
-
dtype=torch.
|
| 40 |
token=HF_TOKEN,
|
| 41 |
# attn_implementation="flash_attention_2",
|
| 42 |
)
|
|
|
|
| 36 |
loaded_models[key] = Qwen3TTSModel.from_pretrained(
|
| 37 |
model_path,
|
| 38 |
device_map="cpu",
|
| 39 |
+
dtype=torch.float32,
|
| 40 |
token=HF_TOKEN,
|
| 41 |
# attn_implementation="flash_attention_2",
|
| 42 |
)
|