Fix uni_pc sampler math. This changes the images this sampler produces.

2023-10-20 04:03:07 -04:00
parent f1062be622
commit 4185324a1d
3 changed files with 33 additions and 24 deletions
--- a/comfy/samplers.py
+++ b/comfy/samplers.py
@@ -739,7 +739,7 @@ class KSampler:
        sigmas = None

        discard_penultimate_sigma = False
-        if self.sampler in ['dpm_2', 'dpm_2_ancestral']:
+        if self.sampler in ['dpm_2', 'dpm_2_ancestral', 'uni_pc', 'uni_pc_bh2']:
            steps += 1
            discard_penultimate_sigma = True