Update models.py (#1116)

2024-05-27 04:12:51 +01:00
parent 90e92b079e
commit 151184598a
1 changed files with 1 additions and 1 deletions
--- a/GPT_SoVITS/module/models.py
+++ b/GPT_SoVITS/module/models.py
@@ -907,7 +907,7 @@ class SynthesizerTrn(nn.Module):
        ge = self.ref_enc(y * y_mask, y_mask)

        with autocast(enabled=False):
-            maybe_no_grad = torch.no_grad() if self.freeze_quantizer else contextlib.nullcontext
+            maybe_no_grad = torch.no_grad() if self.freeze_quantizer else contextlib.nullcontext()
            with maybe_no_grad:
                if self.freeze_quantizer:
                    self.ssl_proj.eval()