Update to work with current ComfyUI

2024-04-14 18:44:33 -07:00
parent d7dc74e7c3
commit 8efe789867
1 changed files with 1 additions and 1 deletions
@@ -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