nodes/ssitu-ComfyUI_restart_sampling
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix assuming first/last steps would be set for non-custom restart samplers

Browse Source
This commit is contained in:
blepping
2024-03-15 19:20:58 -06:00
parent 6ddbaf02f4
commit 20efcba391
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
restart_sampling.py
+2 -2
View File
@@ -93,12 +93,12 @@ def restart_sampling(model, seed, steps, cfg, sampler, scheduler, positive, nega
real_model, model.load_device,
)
if last_step < (len(sigmas) - 1):
if last_step is not None and last_step < (len(sigmas) - 1):
sigmas = sigmas[:last_step + 1]
if force_full_denoise:
sigmas[-1] = 0
if start_step < (len(sigmas) - 1):
if start_step is not None and start_step < (len(sigmas) - 1):
sigmas = sigmas[start_step:]
sampler_wrapper = KSamplerRestartWrapper(sampler, real_model, restart_scheduler, restart_segments, total_steps)