Skip to content

Commit 8dae8d7

Browse files
committed
fix blockwise controlnet training by avoid inplace
1 parent 89bf3ce commit 8dae8d7

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

diffsynth/pipelines/qwen_image.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -761,11 +761,13 @@ def model_fn_qwen_image(
761761
enable_fp8_attention=enable_fp8_attention,
762762
)
763763
if blockwise_controlnet_conditioning is not None:
764-
image[:, :image_seq_len] = image[:, :image_seq_len] + blockwise_controlnet.blockwise_forward(
765-
image=image[:, :image_seq_len], conditionings=blockwise_controlnet_conditioning,
764+
image_slice = image[:, :image_seq_len].clone()
765+
controlnet_output = blockwise_controlnet.blockwise_forward(
766+
image=image_slice, conditionings=blockwise_controlnet_conditioning,
766767
controlnet_inputs=blockwise_controlnet_inputs, block_id=block_id,
767768
progress_id=progress_id, num_inference_steps=num_inference_steps,
768769
)
770+
image[:, :image_seq_len] = image_slice + controlnet_output
769771

770772
image = dit.norm_out(image, conditioning)
771773
image = dit.proj_out(image)

0 commit comments

Comments
 (0)