Update models.py

This commit is contained in:
RVC-Boss 2024-03-14 00:42:15 +08:00 committed by GitHub
parent 3dfbcbcd82
commit c287430ecc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -893,7 +893,7 @@ class SynthesizerTrn(nn.Module):
if freeze_quantizer: if freeze_quantizer:
self.ssl_proj.requires_grad_(False) self.ssl_proj.requires_grad_(False)
self.quantizer.requires_grad_(False) self.quantizer.requires_grad_(False)
self.quantizer.eval() #self.quantizer.eval()
# self.enc_p.text_embedding.requires_grad_(False) # self.enc_p.text_embedding.requires_grad_(False)
# self.enc_p.encoder_text.requires_grad_(False) # self.enc_p.encoder_text.requires_grad_(False)
# self.enc_p.mrte.requires_grad_(False) # self.enc_p.mrte.requires_grad_(False)