stop crashes on non-square images

2024-08-30 20:32:17 +00:00 · 2022-10-25 13:17:06 -04:00
parent dd07392045
commit 4352eb6628
3 changed files with 9 additions and 2 deletions
--- a/ldm/models/diffusion/sampler.py
+++ b/ldm/models/diffusion/sampler.py
@ -265,6 +265,7 @@ class Sampler(object):
            )

            if mask is not None:
+                print('DEBUG: in masking routine')
                assert x0 is not None
                img_orig = self.model.q_sample(
                    x0, ts