Skip to content

Commit 369b079

Browse files
Fix lowvram issue with ltxv vae.
1 parent 9c9a7f0 commit 369b079

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

comfy/ldm/lightricks/vae/causal_video_autoencoder.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -695,7 +695,7 @@ def forward(self, x, causal: bool = True, timestep: Optional[torch.Tensor] = Non
695695
class LayerNorm(nn.Module):
696696
def __init__(self, dim, eps, elementwise_affine=True) -> None:
697697
super().__init__()
698-
self.norm = nn.LayerNorm(dim, eps=eps, elementwise_affine=elementwise_affine)
698+
self.norm = ops.LayerNorm(dim, eps=eps, elementwise_affine=elementwise_affine)
699699

700700
def forward(self, x):
701701
x = rearrange(x, "b c d h w -> b d h w c")

0 commit comments

Comments
 (0)