make sure non-latent diffusion still works

2025-12-20 02:04:19 +01:00 · 2022-04-26 08:36:00 -07:00
parent 0b28ee0d01
commit 4d25976f33
2 changed files with 2 additions and 2 deletions
--- a/dalle2_pytorch/dalle2_pytorch.py
+++ b/dalle2_pytorch/dalle2_pytorch.py
@@ -1117,7 +1117,7 @@ class Decoder(nn.Module):
        unet,
        *,
        clip,
-        vae = None,
+        vae = tuple(),
        timesteps = 1000,
        cond_drop_prob = 0.2,
        loss_type = 'l1',