From 72f33b31221ceab1c612a537dd0e492384b04337 Mon Sep 17 00:00:00 2001 From: Brodie Ferguson <15876037+brodieferguson@users.noreply.github.com> Date: Wed, 8 May 2024 05:55:47 -0600 Subject: [PATCH] fix: impact_sampling - missing UniPC sampler (#592) --- modules/impact/impact_sampling.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/modules/impact/impact_sampling.py b/modules/impact/impact_sampling.py index 8a844ea..d870d10 100644 --- a/modules/impact/impact_sampling.py +++ b/modules/impact/impact_sampling.py @@ -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)