Fix BasicScheduler issue with Loras.

7c9a0f7e · comfyanonymous · 35322a37 · 7c9a0f7e
Commit 7c9a0f7e authored Jan 05, 2024 by comfyanonymous
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 3 deletions

comfy_extras/nodes_custom_sampler.py comfy_extras/nodes_custom_sampler.py +2 -3

No files found.
--- a/comfy_extras/nodes_custom_sampler.py
+++ b/comfy_extras/nodes_custom_sampler.py
@@ -26,9 +26,8 @@ class BasicScheduler:
        if denoise < 1.0:
            total_steps = int(steps/denoise)

-        inner_model = model.patch_model(patch_weights=False)
-        sigmas = comfy.samplers.calculate_sigmas_scheduler(inner_model, scheduler, total_steps).cpu()
-        model.unpatch_model()
+        comfy.model_management.load_models_gpu([model])
+        sigmas = comfy.samplers.calculate_sigmas_scheduler(model.model, scheduler, total_steps).cpu()
        sigmas = sigmas[-(steps + 1):]
        return (sigmas, )