From b5c707ebf21a69f66f38edf231f954496631e22a Mon Sep 17 00:00:00 2001
From: XXXXRT666 <157766680+XXXXRT666@users.noreply.github.com>
Date: Thu, 23 May 2024 02:00:46 +0100
Subject: [PATCH] Update models.py

---
 GPT_SoVITS/module/models.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/GPT_SoVITS/module/models.py b/GPT_SoVITS/module/models.py
index b14e7c81..be0e07c1 100644
--- a/GPT_SoVITS/module/models.py
+++ b/GPT_SoVITS/module/models.py
@@ -906,6 +906,11 @@ 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()
+            with maybe_no_grad:
+                if self.freeze_quantizer:
+                    self.ssl_proj.eval()
+                    self.quantizer.eval()
             ssl = self.ssl_proj(ssl)
             quantized, codes, commit_loss, quantized_list = self.quantizer(
                 ssl, layers=[0]