Update f5_tts/infer/utils_infer.py
Browse files
f5_tts/infer/utils_infer.py
CHANGED
|
@@ -112,9 +112,7 @@ def load_vocoder(vocoder_name="vocos", is_local=False, local_path="", device=dev
|
|
| 112 |
}
|
| 113 |
state_dict.update(encodec_parameters)
|
| 114 |
vocoder.load_state_dict(state_dict)
|
| 115 |
-
vocoder = vocoder.eval()
|
| 116 |
-
vocoder = torch.nn.Module.to_empty(vocoder, device=device)
|
| 117 |
-
#vocoder = vocoder.to_empty(device=device).eval()
|
| 118 |
|
| 119 |
elif vocoder_name == "bigvgan":
|
| 120 |
try:
|
|
|
|
| 112 |
}
|
| 113 |
state_dict.update(encodec_parameters)
|
| 114 |
vocoder.load_state_dict(state_dict)
|
| 115 |
+
vocoder = vocoder.eval().to(device)
|
|
|
|
|
|
|
| 116 |
|
| 117 |
elif vocoder_name == "bigvgan":
|
| 118 |
try:
|