All the unet ops with weights are now handled by comfy.ops

2025-08-02 23:14:49 +08:00 · 2023-12-04 03:12:18 -05:00
parent 6efe561c2a
commit af365e4dd1
4 changed files with 28 additions and 21 deletions
--- a/comfy/ops.py
+++ b/comfy/ops.py
@@ -13,6 +13,14 @@ class Conv3d(torch.nn.Conv3d):
    def reset_parameters(self):
        return None

+class GroupNorm(torch.nn.GroupNorm):
+    def reset_parameters(self):
+        return None
+
+class LayerNorm(torch.nn.LayerNorm):
+    def reset_parameters(self):
+        return None
+
 def conv_nd(dims, *args, **kwargs):
    if dims == 2:
        return Conv2d(*args, **kwargs)