comfyanonymous/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-30 19:31:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

More progress on Sortblock

Browse Source
This commit is contained in:
Jedrzej Kosinski
2025-08-31 20:26:49 -07:00
parent f655fcc5ce
commit cf26d3d58e
2 changed files with 62 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/ldm/flux/model.py
View File

@@ -128,7 +128,7 @@ class Flux(nn.Module):
blocks_replace = patches_replace.get("dit", {})
for i, block in enumerate(self.double_blocks):
transformer_options["block"] = ("double_block", i)
transformer_options["block"] = ("double_block", i, 2)
if ("double_block", i) in blocks_replace:
def block_wrap(args):
out = {}
@@ -170,7 +170,7 @@ class Flux(nn.Module):
img = torch.cat((txt, img), 1)
for i, block in enumerate(self.single_blocks):
transformer_options["block"] = ("single_block", i)
transformer_options["block"] = ("single_block", i, 1)
if ("single_block", i) in blocks_replace:
def block_wrap(args):
out = {}