add gn wrapper and corresponding patcher

lllyasviel · Mar 21, 2024 · 4601658 · 4601658
1 parent 60f5411
commit 4601658
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ldm_patched/ldm/modules/diffusionmodules/openaimodel.py b/ldm_patched/ldm/modules/diffusionmodules/openaimodel.py
@@ -943,7 +943,7 @@ def forward(self, x, timesteps=None, context=None, y=None, control=None, transfo
             h = self.id_predictor(h)
         elif "groupnorm_wrapper" in transformer_options:
             out_norm, out_rest = self.out[0], self.out[1:]
-            h = transformer_options["groupnorm_wrapper"](h, out_norm, transformer_options)
+            h = transformer_options["groupnorm_wrapper"](out_norm, h, transformer_options)
             h = out_rest(h)
         else:
             h = self.out(h)