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

2026-08-29 02:10:59 +08:00 · 2024-07-01 16:52:37 +09:00
parent 7db7d08b5a
commit eb0fb71bfd
1 changed files with 1 additions and 1 deletions
@@ -201,7 +201,7 @@ class DVAE(nn.Module):
            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(