fix: impact_sampling - compatibility patch

comfyanonymous/ComfyUI@1a0486b
ltdrdata · Apr 6, 2024 · 8d4e244 · 8d4e244
1 parent a1694e5
commit 8d4e244
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/modules/impact/config.py b/modules/impact/config.py
@@ -2,7 +2,7 @@
 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
@@ -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:]])