mirror of
https://github.com/RVC-Boss/GPT-SoVITS.git
synced 2025-05-29 03:19:22 +08:00
Update models.py (#1116)
This commit is contained in:
parent
90e92b079e
commit
151184598a
@ -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()
|
||||
|
Loading…
x
Reference in New Issue
Block a user