298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-18 10:03:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

pass transformer options into stream blocks

Browse Source
This commit is contained in:
kabachuha 2025-03-25 20:48:05 +03:00
parent 33528f31be
commit 855ea059a4
1 changed files with 4 additions and 4 deletions

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

@ -131,7 +131,7 @@ class Flux(nn.Module):
txt=args["txt"],
vec=args["vec"],
pe=args["pe"],
attn_mask=args.get("attn_mask"))
attn_mask=args.get("attn_mask"),transformer_options=transformer_options)
return out
out = blocks_replace[("double_block", i)]({"img": img,
@ -147,7 +147,7 @@ class Flux(nn.Module):
txt=txt,
vec=vec,
pe=pe,
attn_mask=attn_mask)
attn_mask=attn_mask, transformer_options=transformer_options)
if control is not None: # Controlnet
control_i = control.get("input")
@ -165,7 +165,7 @@ class Flux(nn.Module):
out["img"] = block(args["img"],
vec=args["vec"],
pe=args["pe"],
attn_mask=args.get("attn_mask"))
attn_mask=args.get("attn_mask"),transformer_options=transformer_options)
return out
out = blocks_replace[("single_block", i)]({"img": img,
@ -175,7 +175,7 @@ class Flux(nn.Module):
{"original_block": block_wrap})
img = out["img"]
else:
img = block(img, vec=vec, pe=pe, attn_mask=attn_mask)
img = block(img, vec=vec, pe=pe, attn_mask=attn_mask, transformer_options=transformer_options)
if control is not None: # Controlnet
control_o = control.get("output")