From abf460fdac0afaec4882b5431f278fdada97838c Mon Sep 17 00:00:00 2001 From: songkey-win Date: Fri, 13 Dec 2024 17:03:18 +0800 Subject: [PATCH] lc --- hellomeme/pipelines/pipline_hm_video.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/hellomeme/pipelines/pipline_hm_video.py b/hellomeme/pipelines/pipline_hm_video.py index 09d1ee6..c1680af 100644 --- a/hellomeme/pipelines/pipline_hm_video.py +++ b/hellomeme/pipelines/pipline_hm_video.py @@ -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: