From 35c0148e79567171bef64d6846e7a62d6b51bcfd Mon Sep 17 00:00:00 2001 From: Jedrzej Kosinski Date: Tue, 16 Jan 2024 17:02:10 -0600 Subject: [PATCH] Made sure autoselect applies USE_EXISTING when no motion models present in Gen2 node --- animatediff/nodes_gen2.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/animatediff/nodes_gen2.py b/animatediff/nodes_gen2.py index b79b828..a4cd9eb 100644 --- a/animatediff/nodes_gen2.py +++ b/animatediff/nodes_gen2.py @@ -62,6 +62,8 @@ class UseEvolvedSamplingNode: # if autoselect, get suggested beta_schedule from motion model if beta_schedule == BetaSchedules.AUTOSELECT and not model.motion_models.is_empty(): beta_schedule = model.motion_models[0].model.get_best_beta_schedule(log=True) + else: + beta_schedule = BetaSchedules.USE_EXISTING new_model_sampling = BetaSchedules.to_model_sampling(beta_schedule, model) if new_model_sampling is not None: model.add_object_patch("model_sampling", new_model_sampling)