Skip to content

Commit 7fb31f0

Browse files
committed
update
1 parent 06b0835 commit 7fb31f0

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

src/diffusers/pipelines/flux/pipeline_flux_controlnet_image_to_image.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -554,6 +554,7 @@ def prepare_latents(
554554
f"You have passed a list of generators of length {len(generator)}, but requested an effective batch"
555555
f" size of {batch_size}. Make sure the batch size matches the length of the generators."
556556
)
557+
557558
# VAE applies 8x compression on images but we must also account for packing which requires
558559
# latent height and width to be divisible by 2.
559560
height = 2 * (int(height) // (self.vae_scale_factor * 2))

tests/pipelines/controlnet_flux/test_controlnet_flux_img2img.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -224,7 +224,7 @@ def test_flux_image_output_shape(self):
224224
pipe = self.pipeline_class(**self.get_dummy_components()).to(torch_device)
225225
inputs = self.get_dummy_inputs(torch_device)
226226

227-
height_width_pairs = [(32, 32), (72, 57)]
227+
height_width_pairs = [(32, 32), (72, 56)]
228228
for height, width in height_width_pairs:
229229
expected_height = height - height % (pipe.vae_scale_factor * 2)
230230
expected_width = width - width % (pipe.vae_scale_factor * 2)

0 commit comments

Comments
 (0)