From 69f10d384798c24aedee40c380fee1ecf9daa618 Mon Sep 17 00:00:00 2001 From: aurel-g Date: Wed, 1 Jan 2025 15:54:24 +0100 Subject: [PATCH] Fix calculate_sigmas import error --- keyframe/util.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/keyframe/util.py b/keyframe/util.py index cff4d1a..e5afa14 100644 --- a/keyframe/util.py +++ b/keyframe/util.py @@ -2,7 +2,7 @@ import math import torch -from comfy.samplers import calculate_sigmas_scheduler +from comfy.samplers import calculate_sigmas KEYFRAME_INJECTED_ATTR = "keyframe_injected" @@ -40,10 +40,10 @@ def generate_sigmas(real_model, x, origin_sigmas, scheduler, steps, part_group, for part in part_group: if part.denoise is None or part.denoise > 0.9999: - new_sigmas[part.batch_index] = calculate_sigmas_scheduler(real_model, scheduler, steps).to(device) + new_sigmas[part.batch_index] = calculate_sigmas(real_model, scheduler, steps).to(device) else: new_steps = int(steps / part.denoise) - sigmas = calculate_sigmas_scheduler(real_model, scheduler, new_steps).to(device) + sigmas = calculate_sigmas(real_model, scheduler, new_steps).to(device) new_sigmas[part.batch_index] = sigmas[-(steps + 1):] return new_sigmas