fix(dvae): unnecessary memory consumption (fix #148)

2noise · Jul 1, 2024 · eb0fb71 · eb0fb71
1 parent 7db7d08
commit eb0fb71
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ChatTTS/model/dvae.py b/ChatTTS/model/dvae.py
@@ -201,7 +201,7 @@ def forward(self, inp: torch.Tensor) -> torch.Tensor:
             if self.vq_layer is not None:
                 vq_feats = self.vq_layer._embed(inp)
             else:
-                vq_feats = inp.detach().clone()
+                vq_feats = inp
 
             vq_feats = (
                 vq_feats.view(