Skip to content

Commit 8bdb806

Browse files
committed
update
1 parent 96910d0 commit 8bdb806

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/pipelines/chroma/pipeline_chroma_img2img.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -802,7 +802,7 @@ def __call__(
802802

803803
# 4. Prepare timesteps
804804
sigmas = np.linspace(1.0, 1 / num_inference_steps, num_inference_steps) if sigmas is None else sigmas
805-
image_seq_len = latents.shape[1]
805+
image_seq_len = (int(height) // self.vae_scale_factor // 2) * (int(width) // self.vae_scale_factor // 2)
806806
mu = calculate_shift(
807807
image_seq_len,
808808
self.scheduler.config.get("base_image_seq_len", 256),

0 commit comments

Comments
 (0)