support sparsectrl

2026-04-03 09:46:36 +02:00 · 2023-12-15 20:55:51 +08:00
parent 6c8a01b148
commit 401bc45697
7 changed files with 697 additions and 45 deletions
--- a/animatediff/models/resnet.py
+++ b/animatediff/models/resnet.py
@@ -123,7 +123,7 @@ class ResnetBlock3D(nn.Module):
        time_embedding_norm="default",
        output_scale_factor=1.0,
        use_in_shortcut=None,
-        use_inflated_groupnorm=None,
+        use_inflated_groupnorm=False,
    ):
        super().__init__()
        self.pre_norm = pre_norm