Skip to content

Commit 7ffbc25

Browse files
Update train_controlnet_flux.py,Fix size mismatch issue in validation (#9679)
Update train_controlnet_flux.py Fix the problem of inconsistency between size of image and size of validation_image which causes np.stack to report error. Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
1 parent f55f1f7 commit 7ffbc25

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

examples/controlnet/train_controlnet_flux.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -152,6 +152,7 @@ def log_validation(
152152
guidance_scale=3.5,
153153
generator=generator,
154154
).images[0]
155+
image = image.resize((args.resolution, args.resolution))
155156
images.append(image)
156157
image_logs.append(
157158
{"validation_image": validation_image, "images": images, "validation_prompt": validation_prompt}

0 commit comments

Comments
 (0)