From bc316246577f5076ce96de4603360c6155e9a880 Mon Sep 17 00:00:00 2001
From: ScilenceForest <45549187+ScilenceForest@users.noreply.github.com>
Date: Tue, 15 Oct 2024 16:13:36 +0800
Subject: [PATCH] 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.
---
 examples/controlnet/train_controlnet_flux.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/examples/controlnet/train_controlnet_flux.py b/examples/controlnet/train_controlnet_flux.py
index 5969218f3c3e..ad01b8a39c36 100644
--- a/examples/controlnet/train_controlnet_flux.py
+++ b/examples/controlnet/train_controlnet_flux.py
@@ -152,6 +152,7 @@ def log_validation(
                     guidance_scale=3.5,
                     generator=generator,
                 ).images[0]
+            image = image.resize((args.resolution, args.resolution))
             images.append(image)
         image_logs.append(
             {"validation_image": validation_image, "images": images, "validation_prompt": validation_prompt}