mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-03-15 05:57:20 +00:00
Fix lowvram issue with ltxv vae.
This commit is contained in:
parent
9c9a7f012a
commit
369b079ff6
@ -695,7 +695,7 @@ class DepthToSpaceUpsample(nn.Module):
|
|||||||
class LayerNorm(nn.Module):
|
class LayerNorm(nn.Module):
|
||||||
def __init__(self, dim, eps, elementwise_affine=True) -> None:
|
def __init__(self, dim, eps, elementwise_affine=True) -> None:
|
||||||
super().__init__()
|
super().__init__()
|
||||||
self.norm = nn.LayerNorm(dim, eps=eps, elementwise_affine=elementwise_affine)
|
self.norm = ops.LayerNorm(dim, eps=eps, elementwise_affine=elementwise_affine)
|
||||||
|
|
||||||
def forward(self, x):
|
def forward(self, x):
|
||||||
x = rearrange(x, "b c d h w -> b d h w c")
|
x = rearrange(x, "b c d h w -> b d h w c")
|
||||||
|
Loading…
Reference in New Issue
Block a user