chore: speck (#456)

* chore: speck * chore: speck
fishaudio · Aug 8, 2024 · a4bacb7 · a4bacb7
1 parent de565f7
commit a4bacb7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/fish_speech/models/vqgan/modules/firefly.py b/fish_speech/models/vqgan/modules/firefly.py
@@ -242,7 +242,7 @@ def forward(self, x, template=None):
             if self.use_template:
                 x = x + self.noise_convs[i](template)
 
-            if self.training and self.checkpointing:
+            if self.training:
                 x = checkpoint(
                     self.resblocks[i],
                     x,
@@ -512,7 +512,7 @@ def forward(self, x: torch.Tensor, template=None, mask=None) -> torch.Tensor:
         if x.ndim == 2:
             x = x[:, None, :]
 
-        if self.vq is not None:
+        if self.quantizer is not None:
             return x, vq_result
 
         return x