wuyingwen/comfyui
1
0
Fork 0
Code Issues Pull Requests Actions 5 Packages Projects Releases Wiki Activity

Fix BasicScheduler issue with Loras.

Browse Source
This commit is contained in:
comfyanonymous 2024-01-05 12:31:13 -05:00
parent 35322a3766
commit 7c9a0f7e0a
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy_extras/nodes_custom_sampler.py
View File

@ -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, )