Update model.py

model.py

@@ -466,7 +466,6 @@ class StreamMultiDiffusion(nn.Module):
             self.state['background'].latent = (
                 self.encode_imgs(T.ToTensor()(image_)[None].to(self.device, self.dtype))
             )  # (1, 3, H, W)
-            print(self.state['background'].latent.dtype, self.state['background'].latent.device, self.vae.dtype, self.vae.device)
             self.state['background'].prompt = prompt
             self.state['background'].negative_prompt = negative_prompt
             self.state['background'].embed = embed
@@ -475,6 +474,7 @@ class StreamMultiDiffusion(nn.Module):
                 if self.white is None:
                     self.white = self.encode_imgs(torch.ones(1, 3, self.height, self.width, dtype=self.dtype, device=self.device))
                 mix_ratio = self.bootstrap_mix_ratios[:, None, None, None]
+                print(mix_ratio, mix_ratio.dtype, self.white.dtype, self.white.device, self.state['background'].latent.dtype, self.state['background'].latent.device)
                 self.bootstrap_latent = mix_ratio * self.white + (1.0 - mix_ratio) * self.state['background'].latent
 
             self.ready_checklist['background_registered'] = True