asagi4 / comfyui-prompt-control

ComfyUI nodes for prompt editing and LoRA control

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

SamplerCustom is referencing cfg as the step count when scheduling Lora

pyomyofiro opened this issue · comments

SamplerCustom is referencing cfg as the step count when scheduling Lora
rapture_20240319225456
rapture_20240319225605
rapture_20240319225649
rapture_20240319225725

When I want to use SamplerCustom, I am manually altering this part to the actual step count value.
rapture_20240319225932

Hm. Looks like the sample_custom function needs a completely different hijack, since the steps value isn't even available.

I never use the custom sampler nodes, so I never noticed this.

The above commit isn't a perfect fix since some samplers discard the penultimate sigma, but it's close enough that I'll consider this issue fixed.