fix support for ComfyUI_PuLID_Flux_ll

chengzeyi · Jan 15, 2025 · 36ba3c8 · 36ba3c8
1 parent a7caa12
commit 36ba3c8
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/first_block_cache.py b/first_block_cache.py
@@ -565,7 +565,7 @@ def create_patch_flux_forward_orig(model,
     from comfy.ldm.flux.model import timestep_embedding
 
     def call_remaining_blocks(self, blocks_replace, control, img, txt, vec, pe,
-                              attn_mask, ca_idx, timesteps):
+                              attn_mask, ca_idx, timesteps, transformer_options):
         original_hidden_states = img
 
         extra_block_forward_kwargs = {}
@@ -595,7 +595,8 @@ def block_wrap(args):
                                           "pe": pe,
                                           **extra_block_forward_kwargs
                                       }, {
-                                          "original_block": block_wrap
+                                          "original_block": block_wrap,
+                                          "transformer_options": transformer_options
                                       })
                 txt = out["txt"]
                 img = out["img"]
@@ -644,7 +645,8 @@ def block_wrap(args):
                                           "pe": pe,
                                           **extra_block_forward_kwargs
                                       }, {
-                                          "original_block": block_wrap
+                                          "original_block": block_wrap,
+                                          "transformer_options": transformer_options
                                       })
                 img = out["img"]
             else:
@@ -741,7 +743,8 @@ def block_wrap(args):
                                           "pe": pe,
                                           **extra_block_forward_kwargs
                                       }, {
-                                          "original_block": block_wrap
+                                          "original_block": block_wrap,
+                                          "transformer_options": transformer_options
                                       })
                 txt = out["txt"]
                 img = out["img"]
@@ -799,6 +802,7 @@ def block_wrap(args):
                 attn_mask,
                 ca_idx,
                 timesteps,
+                transformer_options,
             )
             set_buffer("hidden_states_residual", hidden_states_residual)
         torch._dynamo.graph_break()

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 name = "wavespeed"
 description = "The all in one inference optimization solution for ComfyUI, universal, flexible, and fast."
-version = "1.1.6"
+version = "1.1.7"
 license = {file = "LICENSE"}
 
 [project.urls]