Update f5_tts/infer/utils_infer.py

f5_tts/infer/utils_infer.py

@@ -112,9 +112,9 @@ def load_vocoder(vocoder_name="vocos", is_local=False, local_path="", device=dev
             }
             state_dict.update(encodec_parameters)
         vocoder.load_state_dict(state_dict)
-        #vocoder = vocoder.eval().to(device)
-        #vocoder = torch.nn.Module.to_empty(vocoder, device=device)
-        vocoder = vocoder.to_empty(device=device).eval()
+        vocoder = vocoder.eval()
+        vocoder = torch.nn.Module.to_empty(vocoder, device=device)
+        #vocoder = vocoder.to_empty(device=device).eval()
         
     elif vocoder_name == "bigvgan":
         try: