reformat

staoxiao · staoxiao · commit 5f3148d801e6 · 2024-12-08T16:31:52.000+08:00
diff --git a/src/diffusers/pipelines/lumina/pipeline_lumina.py b/src/diffusers/pipelines/lumina/pipeline_lumina.py
@@ -777,7 +777,7 @@ def __call__(
 
         # 4. Prepare timesteps
         timesteps, num_inference_steps = retrieve_timesteps(
-            self.scheduler, num_inference_steps, device, timesteps,
+            self.scheduler, num_inference_steps, device, timesteps, sigmas
         )
 
         # 5. Prepare latents.
diff --git a/src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py b/src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py
@@ -300,7 +300,7 @@ def step(
 
         sigma = self.sigmas[self.step_index]
         sigma_next = self.sigmas[self.step_index + 1]
-        
+
         prev_sample = sample + (sigma_next - sigma) * model_output
 
         # Cast sample back to model compatible dtype

Original file line number	Diff line number	Diff line change
`@@ -777,7 +777,7 @@ def __call__(`
`777`	`777`
`778`	`778`	`# 4. Prepare timesteps`
`779`	`779`	`timesteps, num_inference_steps = retrieve_timesteps(`
`780`		`- self.scheduler, num_inference_steps, device, timesteps,`
	`780`	`+ self.scheduler, num_inference_steps, device, timesteps, sigmas`
`781`	`781`	`)`
`782`	`782`
`783`	`783`	`# 5. Prepare latents.`