diff --git a/modules/impact/config.py b/modules/impact/config.py index 0e05b10..c521008 100644 --- a/modules/impact/config.py +++ b/modules/impact/config.py @@ -2,7 +2,7 @@ import configparser import os -version_code = [4, 86] +version_code = [4, 86, 1] version = f"V{version_code[0]}.{version_code[1]}" + (f'.{version_code[2]}' if len(version_code) > 2 else '') dependency_version = 20 diff --git a/modules/impact/impact_sampling.py b/modules/impact/impact_sampling.py index 8ff6f52..78bf361 100644 --- a/modules/impact/impact_sampling.py +++ b/modules/impact/impact_sampling.py @@ -13,7 +13,11 @@ def calculate_sigmas(model, sampler, scheduler, steps): steps += 1 discard_penultimate_sigma = True - sigmas = samplers.calculate_sigmas_scheduler(model.model, scheduler, steps) + if hasattr(samplers, 'calculate_sigmas'): + sigmas = samplers.calculate_sigmas(model.get_model_object("model_sampling"), scheduler, steps) + else: + print(f"[Impact Pack] calculate_sigmas: ComfyUI is an outdated version.") + sigmas = samplers.calculate_sigmas_scheduler(model.model, scheduler, steps) if discard_penultimate_sigma: sigmas = torch.cat([sigmas[:-2], sigmas[-1:]])