theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-18 22:08:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

update controlnet_frames, downloads

Browse Source
This commit is contained in:
tc2000731
2024-10-31 17:38:57 +08:00
parent 900a1c095f
commit 9377214518
2 changed files with 39 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
diffsynth/pipelines/flux_image.py
View File

@@ -258,6 +258,7 @@ def lets_dance_flux(
):
if tiled:
def flux_forward_fn(hl, hr, wl, wr):
tiled_controlnet_frames = [f[:, :, hl: hr, wl: wr] for f in controlnet_frames] if controlnet_frames is not None else None
return lets_dance_flux(
dit=dit,
controlnet=controlnet,
@@ -268,7 +269,7 @@ def lets_dance_flux(
guidance=guidance,
text_ids=text_ids,
image_ids=None,
controlnet_frames=[f[:, :, hl: hr, wl: wr] for f in controlnet_frames],
controlnet_frames=tiled_controlnet_frames,
tiled=False,
**kwargs
)