LTXV lowvram fixes.

2024-11-22 17:17:11 -05:00
parent bc6be6c11e
commit e5c3f4b87f
4 changed files with 13 additions and 10 deletions
--- a/comfy/ldm/lightricks/vae/causal_conv3d.py
+++ b/comfy/ldm/lightricks/vae/causal_conv3d.py
@@ -2,6 +2,8 @@ from typing import Tuple, Union

 import torch
 import torch.nn as nn
+import comfy.ops
+ops = comfy.ops.disable_weight_init


 class CausalConv3d(nn.Module):
@@ -29,7 +31,7 @@ class CausalConv3d(nn.Module):
        width_pad = kernel_size[2] // 2
        padding = (0, height_pad, width_pad)

-        self.conv = nn.Conv3d(
+        self.conv = ops.Conv3d(
            in_channels,
            out_channels,
            kernel_size,