From d83a7af4db9d2f7d1aa6ca326f462c4f634a56ab Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Fri, 5 Jul 2024 10:52:40 +0300 Subject: [PATCH] better compatibility with older diffusers --- mimicmotion/pipelines/pipeline_mimicmotion.py | 5 ++++- pyproject.toml | 2 +- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/mimicmotion/pipelines/pipeline_mimicmotion.py b/mimicmotion/pipelines/pipeline_mimicmotion.py index 40245ae..4938481 100644 --- a/mimicmotion/pipelines/pipeline_mimicmotion.py +++ b/mimicmotion/pipelines/pipeline_mimicmotion.py @@ -567,7 +567,10 @@ class MimicMotionPipeline(DiffusionPipeline): added_time_ids = added_time_ids.to(device) # 6. Prepare timesteps - timesteps, num_inference_steps = retrieve_timesteps(self.scheduler, num_inference_steps, device, None, sigmas) + if sigmas is not None: + timesteps, num_inference_steps = retrieve_timesteps(self.scheduler, num_inference_steps, device, None, sigmas) + else: + timesteps, num_inference_steps = retrieve_timesteps(self.scheduler, num_inference_steps, device, None) # 5. Prepare latent variables num_channels_latents = self.unet.config.in_channels diff --git a/pyproject.toml b/pyproject.toml index 92a995d..c14acf6 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,7 +1,7 @@ [project] name = "comfyui-mimicmotionwrapper" description = "Optimized wrapper nodes for MimicMotion: [a/https://github.com/tencent/MimicMotion](https://github.com/tencent/MimicMotion)" -version = "1.0.0" +version = "1.0.1" license = "Apache-2.0" dependencies = ["diffusers>=0.27.0", "transformers>=4.32.1"]