better compatibility with older diffusers

This commit is contained in:
kijai
2024-07-05 10:52:40 +03:00
parent 6bed431839
commit d83a7af4db
2 changed files with 5 additions and 2 deletions
@@ -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
+1 -1
View File
@@ -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"]