Spaces:

Gregniuki
/

f5-tts_Polish_English_German

Running on Zero

Gregniuki commited on Nov 28, 2024

Commit

89d38a4

verified ·

1 Parent(s): c4fc213

Update infer/utils_infer.py

Files changed (1) hide show

infer/utils_infer.py CHANGED Viewed

@@ -116,7 +116,7 @@ def load_vocoder(vocoder_name="vocos", is_local=False, local_path="", device=dev
         vocoder.load_state_dict(state_dict)
         # Convert vocoder to bfloat16 if using a compatible device
-        vocoder = vocoder.eval().to(device).to(torch.float32)
     elif vocoder_name == "bigvgan":
         try:

         vocoder.load_state_dict(state_dict)
         # Convert vocoder to bfloat16 if using a compatible device
+        vocoder = vocoder.eval().to(device).to(torch.float16)
     elif vocoder_name == "bigvgan":
         try: