github bot update + heunpp2

2026-08-16 13:13:16 +02:00 · 2023-11-15 01:36:14 -08:00
parent 861c8d38df
commit a9bd188555
9 changed files with 161 additions and 59 deletions
@@ -624,6 +624,11 @@ class UNetModel(nn.Module):
            transformer_options["block"] = ("input", id)
            h = forward_timestep_embed(module, h, emb, context, transformer_options)
            h = apply_control(h, control, 'input')
+            if "input_block_patch" in transformer_patches:
+                patch = transformer_patches["input_block_patch"]
+                for p in patch:
+                    h = p(h, transformer_options)
+
            hs.append(h)

        transformer_options["block"] = ("middle", 0)