diff --git a/nodes.py b/nodes.py index f7e3aa7..dc480c2 100644 --- a/nodes.py +++ b/nodes.py @@ -263,7 +263,7 @@ class MimicMotionSampler: mm.soft_empty_cache() dtype = mimic_pipeline['dtype'] pipeline = mimic_pipeline['pipeline'] - if "1-1" in mimic_pipeline['model_name'] and context_size is not 72: + if "1-1" in mimic_pipeline['model_name'] and context_size != 72: log.warning("WARNING: 1.1 model should use 72 frame context_lenght") original_scheduler = pipeline.scheduler diff --git a/pyproject.toml b/pyproject.toml index 924e515..d0c3578 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.1" +version = "1.0.2" license = { text = "Apache License 2.0" } dependencies = ["diffusers>=0.27.0", "transformers>=4.32.1"]