Spaces:

lshzhm
/

DeepAudio-V1

Running

lshzhm commited on 8 days ago

Commit

e33b81a

1 Parent(s): 5f78c7f

Update utils_infer.py

Files changed (1) hide show

F5-TTS/src/f5_tts/infer/utils_infer.py CHANGED Viewed

@@ -264,7 +264,7 @@ def load_model(
     ).to(device)
     #dtype = torch.float32 if mel_spec_type == "bigvgan" else None
-    dtype = torch.float32
     model = load_checkpoint(model, ckpt_path, device, dtype=dtype, use_ema=use_ema)
     return model

     ).to(device)
     #dtype = torch.float32 if mel_spec_type == "bigvgan" else None
+    dtype = torch.float16
     model = load_checkpoint(model, ckpt_path, device, dtype=dtype, use_ema=use_ema)
     return model