latents.shape -> latents_shape

skirsten · skirsten · commit da3311d2ff7c · 2022-12-29T13:28:31.000+01:00
diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_flax_stable_diffusion_img2img.py b/src/diffusers/pipelines/stable_diffusion/pipeline_flax_stable_diffusion_img2img.py
@@ -258,7 +258,7 @@ def loop_body(step, args):
             return latents, scheduler_state
 
         scheduler_state = self.scheduler.set_timesteps(
-            params["scheduler"], num_inference_steps=num_inference_steps, shape=latents.shape
+            params["scheduler"], num_inference_steps=num_inference_steps, shape=latents_shape
         )
 
         latent_timestep = scheduler_state.timesteps[start_timestep : start_timestep + 1].repeat(batch_size)

Original file line number	Diff line number	Diff line change
`@@ -258,7 +258,7 @@ def loop_body(step, args):`
`258`	`258`	`return latents, scheduler_state`
`259`	`259`
`260`	`260`	`scheduler_state = self.scheduler.set_timesteps(`
`261`		`- params["scheduler"], num_inference_steps=num_inference_steps, shape=latents.shape`
	`261`	`+ params["scheduler"], num_inference_steps=num_inference_steps, shape=latents_shape`
`262`	`262`	`)`
`263`	`263`
`264`	`264`	`latent_timestep = scheduler_state.timesteps[start_timestep : start_timestep + 1].repeat(batch_size)`