JymNils commited on
Commit
d92759a
·
verified ·
1 Parent(s): 1568b79

Update app.py

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