Sampling code refactor.

2023-09-26 13:45:15 -04:00
parent aeba1cc2a0
commit 446caf711c
2 changed files with 150 additions and 113 deletions
--- a/comfy/ldm/models/diffusion/ddim.py
+++ b/comfy/ldm/models/diffusion/ddim.py
@@ -59,7 +59,7 @@ class DDIMSampler(object):
    @torch.no_grad()
    def sample_custom(self,
                      ddim_timesteps,
-                      conditioning,
+                      conditioning=None,
                      callback=None,
                      img_callback=None,
                      quantize_x0=False,