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)