fix shape

linoytsaban · linoytsaban · commit 070a9856ab09 · 2025-05-26T22:32:08.000+03:00
diff --git a/src/diffusers/pipelines/sana/pipeline_sana_sprint_img2img.py b/src/diffusers/pipelines/sana/pipeline_sana_sprint_img2img.py
@@ -860,7 +860,8 @@ def __call__(
                 f"After adjusting the num_inference_steps by strength parameter: {strength}, the number of pipeline"
                 f"steps is {num_inference_steps} which is < 1 and not appropriate for this pipeline."
             )
-        latent_timestep = timesteps[:1].repeat(batch_size * num_images_per_prompt)
+        #latent_timestep = timesteps[:1].repeat(batch_size * num_images_per_prompt)
+        latent_timestep = timesteps[:1]
 
         # 5. Prepare latents.
         latent_channels = self.transformer.config.in_channels

Original file line number	Diff line number	Diff line change
`@@ -860,7 +860,8 @@ def __call__(`
`860`	`860`	`f"After adjusting the num_inference_steps by strength parameter: {strength}, the number of pipeline"`
`861`	`861`	`f"steps is {num_inference_steps} which is < 1 and not appropriate for this pipeline."`
`862`	`862`	`)`
`863`		`- latent_timestep = timesteps[:1].repeat(batch_size * num_images_per_prompt)`
	`863`	`+ #latent_timestep = timesteps[:1].repeat(batch_size * num_images_per_prompt)`
	`864`	`+ latent_timestep = timesteps[:1]`
`864`	`865`
`865`	`866`	`# 5. Prepare latents.`
`866`	`867`	`latent_channels = self.transformer.config.in_channels`