diff --git a/GPT_SoVITS/module/models.py b/GPT_SoVITS/module/models.py index a4d2235..0a49513 100644 --- a/GPT_SoVITS/module/models.py +++ b/GPT_SoVITS/module/models.py @@ -893,6 +893,7 @@ class SynthesizerTrn(nn.Module): if freeze_quantizer: self.ssl_proj.requires_grad_(False) self.quantizer.requires_grad_(False) + self.quentizer.eval() # self.enc_p.text_embedding.requires_grad_(False) # self.enc_p.encoder_text.requires_grad_(False) # self.enc_p.mrte.requires_grad_(False)