nodes/BlakeOne-ComfyUI-SchedulerMixer
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Conform with latest ComfyUI updates.py

Browse Source
This commit is contained in:
BlakeOne
2024-04-14 18:53:56 -07:00
committed by GitHub
parent 8efe789867
commit 2771ff9fc5
1 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
node.py
+6 -7
View File
@@ -24,21 +24,20 @@ class SchedulerMixer:
FUNCTION = "get_sigmas"
def get_sigmas(self, model, steps, denoise, normal, karras, exponential, sgm_uniform, simple, ddim_uniform):
total_steps = steps
total_steps = steps
if denoise < 1.0:
total_steps = int(steps / denoise)
comfy.model_management.load_models_gpu([model])
if denoise <= 0.0:
return (torch.FloatTensor([]),)
total_steps = int(steps/denoise)
scheduler_weights = [normal, karras, exponential, sgm_uniform, simple, ddim_uniform]
scheduler_names = ["normal", "karras", "exponential", "sgm_uniform", "simple", "ddim_uniform"]
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(model.model, name, total_steps).cpu()
if weight > 0.0:
sigmas = comfy.samplers.calculate_sigmas(model.get_model_object("model_sampling"), name, total_steps).cpu()
sigmas = sigmas[-(steps + 1):]
mixed_sigmas += sigmas * weight
return (mixed_sigmas,)