mirror of
https://github.com/wassname/TTS.git
synced 2026-10-29 12:51:00 +08:00
Use torch.no_grad for VITS inference
This commit is contained in:
1 parent
3f03e3012c
commit
5021a03de0
1 file changed
+1
@@ -982,6 +982,7 @@ class Vits(BaseTTS):
|
||||
return aux_input["x_lengths"]
|
||||
return torch.tensor(x.shape[1:2]).to(x.device)
|
||||
|
||||
@torch.no_grad()
|
||||
def inference(
|
||||
self, x, aux_input={"x_lengths": None, "d_vectors": None, "speaker_ids": None, "language_ids": None}
|
||||
): # pylint: disable=dangerous-default-value
|
||||
|
||||
Reference in new issue
Block a user