Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "d6099f1078881eb98c8557683e4160eb725bfb8a"
Commit 7c9a0f7e authored by comfyanonymous's avatar comfyanonymous
Browse files

Fix BasicScheduler issue with Loras.

parent 35322a37
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
comfy_extras/nodes_custom_sampler.py
View file @ 7c9a0f7e
...@@ -26,9 +26,8 @@ class BasicScheduler: ...@@ -26,9 +26,8 @@ class BasicScheduler:
if denoise < 1.0: if denoise < 1.0:
total_steps = int(steps/denoise) total_steps = int(steps/denoise)
inner_model = model.patch_model(patch_weights=False) comfy.model_management.load_models_gpu([model])
sigmas = comfy.samplers.calculate_sigmas_scheduler(inner_model, scheduler, total_steps).cpu() sigmas = comfy.samplers.calculate_sigmas_scheduler(model.model, scheduler, total_steps).cpu()
model.unpatch_model()
sigmas = sigmas[-(steps + 1):] sigmas = sigmas[-(steps + 1):]
return (sigmas, ) return (sigmas, )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment