blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-10-06 22:50:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update models.py (#1116)

Browse Source
This commit is contained in:
XXXXRT666 2024-05-27 04:12:51 +01:00 committed by GitHub
parent d877c4e40c
commit 6d8017f724
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
GPT_SoVITS/module/models.py
View File

@ -907,7 +907,7 @@ class SynthesizerTrn(nn.Module):
ge = self.ref_enc(y * y_mask, y_mask) ge = self.ref_enc(y * y_mask, y_mask)
with autocast(enabled=False): 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: with maybe_no_grad:
if self.freeze_quantizer: if self.freeze_quantizer:
self.ssl_proj.eval() self.ssl_proj.eval()