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)