added missing initialization of latent_noise to None

cognidesign · Sep 8, 2022 · 4168774 · 4168774
1 parent d1d044a
commit 4168774
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ldm/dream/generator/inpaint.py b/ldm/dream/generator/inpaint.py
@@ -11,6 +11,7 @@
 
 class Inpaint(Img2Img):
     def __init__(self,model):
+        self.init_latent = None
         super().__init__(model)
 
     @torch.no_grad()