nodes/ltdrdata-ComfyUI-Impact-Pack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: impact_sampling - missing UniPC sampler (#592)

Browse Source
This commit is contained in:
Brodie Ferguson
2024-05-08 20:55:47 +09:00
committed by GitHub
parent c16066ecd5
commit 72f33b3122
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/impact/impact_sampling.py
+6 -1
View File
@@ -99,8 +99,13 @@ def ksampler(sampler_name, total_sigmas, extra_options={}, inpaint_options={}):
return k_diffusion_sampling.sample_dpmpp_3m_sde_gpu(model, x, sigmas, **kwargs)
sampler_function = sample_dpmpp_sde
else:
if sampler_name == 'ddim':
if sampler_name == "uni_pc":
return samplers.KSAMPLER(samplers.uni_pc.sample_unipc)
elif sampler_name == "uni_pc_bh2":
return samplers.KSAMPLER(samplers.uni_pc.sample_unipc_bh2)
elif sampler_name == 'ddim':
return samplers.ksampler("euler", inpaint_options={"random": True})
else:
return samplers.ksampler(sampler_name, extra_options, inpaint_options)