lc

2024-12-13 17:03:18 +08:00
parent 0924445042
commit abf460fdac
1 changed files with 1 additions and 1 deletions
@@ -293,7 +293,7 @@ class HMVideoPipeline(StableDiffusionImg2ImgPipeline):
                            )

                tmp_timesteps, tmp_steps = retrieve_timesteps(
-                    scheduler, 10, device, timesteps, sigmas
+                    scheduler, min(10, num_inference_steps), device, timesteps, sigmas
                )

                if idx == 0: