From 9808bd4f5b0f6e65955508c3a5dac6c18a75f35d Mon Sep 17 00:00:00 2001 From: Jedrzej Kosinski Date: Wed, 1 Nov 2023 02:37:42 -0500 Subject: [PATCH] Updated CheckpointLoaderSimpleWithNoiseSelect to work with new ComfyUI changes --- animatediff/nodes.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/animatediff/nodes.py b/animatediff/nodes.py index b28ee78..698f729 100644 --- a/animatediff/nodes.py +++ b/animatediff/nodes.py @@ -537,8 +537,7 @@ class CheckpointLoaderSimpleWithNoiseSelect: ckpt_path = folder_paths.get_full_path("checkpoints", ckpt_name) out = load_checkpoint_guess_config(ckpt_path, output_vae=True, output_clip=True, embedding_directory=folder_paths.get_folder_paths("embeddings")) # register chosen beta schedule on model - convert to beta_schedule name recognized by ComfyUI - beta_schedule_name = BetaSchedules.to_name(beta_schedule) - out[0].model.register_schedule(given_betas=None, beta_schedule=beta_schedule_name, timesteps=1000, linear_start=0.00085, linear_end=0.012, cosine_s=8e-3) + out[0].model.model_sampling = BetaSchedules.to_model_sampling(beta_schedule, out[0]) return out