wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-12-18 10:33:01 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Fixed FlipFlipModule.execute_blocks having hardcoded strings from Qwen

Browse Source
This commit is contained in:
Jedrzej Kosinski 2025-10-02 17:49:43 -07:00
parent 6d3ec9fcf3
commit 8d7b22b720
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/ldm/flipflop_transformer.py
View File

@ -56,8 +56,8 @@ class FlipFlopModule(torch.nn.Module):
out = func(i, block, *out, *args, **kwargs) out = func(i, block, *out, *args, **kwargs)
if isinstance(out, torch.Tensor): if isinstance(out, torch.Tensor):
out = (out,) out = (out,)
if "transformer_blocks" in self.flipflop: if block_type in self.flipflop:
holder = self.flipflop["transformer_blocks"] holder = self.flipflop[block_type]
with holder.context() as ctx: with holder.context() as ctx:
for i, block in enumerate(holder.blocks): for i, block in enumerate(holder.blocks):
out = ctx(func, i, block, *out, *args, **kwargs) out = ctx(func, i, block, *out, *args, **kwargs)