From 8efe78986788ccf68fbda44bcad9d6b178cb0512 Mon Sep 17 00:00:00 2001
From: BlakeOne <30273164+BlakeOne@users.noreply.github.com>
Date: Sun, 14 Apr 2024 18:44:33 -0700
Subject: [PATCH] Update to work with current ComfyUI

---
 node.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/node.py b/node.py
index 167cee4..cbcea2f 100644
--- a/node.py
+++ b/node.py
@@ -36,7 +36,7 @@ class SchedulerMixer:
         mixed_sigmas = torch.zeros((steps + 1,), device="cpu", dtype=torch.float)
         for weight, name in zip(scheduler_weights, scheduler_names):                        
             if weight > 0.0:                
-                sigmas = comfy.samplers.calculate_sigmas_scheduler(model.model, name, total_steps).cpu()
+                sigmas = comfy.samplers.calculate_sigmas(model.model, name, total_steps).cpu()
                 sigmas = sigmas[-(steps + 1):]                
                 mixed_sigmas += sigmas * weight