ironjr commited on
Commit
11ad5ee
·
verified ·
1 Parent(s): 0fcbdc2

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -466,7 +466,6 @@ class StreamMultiDiffusion(nn.Module):
466
  self.state['background'].latent = (
467
  self.encode_imgs(T.ToTensor()(image_)[None].to(self.device, self.dtype))
468
  ) # (1, 3, H, W)
469
- print(self.state['background'].latent.dtype, self.state['background'].latent.device, self.vae.dtype, self.vae.device)
470
  self.state['background'].prompt = prompt
471
  self.state['background'].negative_prompt = negative_prompt
472
  self.state['background'].embed = embed
@@ -475,6 +474,7 @@ class StreamMultiDiffusion(nn.Module):
475
  if self.white is None:
476
  self.white = self.encode_imgs(torch.ones(1, 3, self.height, self.width, dtype=self.dtype, device=self.device))
477
  mix_ratio = self.bootstrap_mix_ratios[:, None, None, None]
 
478
  self.bootstrap_latent = mix_ratio * self.white + (1.0 - mix_ratio) * self.state['background'].latent
479
 
480
  self.ready_checklist['background_registered'] = True
 
466
  self.state['background'].latent = (
467
  self.encode_imgs(T.ToTensor()(image_)[None].to(self.device, self.dtype))
468
  ) # (1, 3, H, W)
 
469
  self.state['background'].prompt = prompt
470
  self.state['background'].negative_prompt = negative_prompt
471
  self.state['background'].embed = embed
 
474
  if self.white is None:
475
  self.white = self.encode_imgs(torch.ones(1, 3, self.height, self.width, dtype=self.dtype, device=self.device))
476
  mix_ratio = self.bootstrap_mix_ratios[:, None, None, None]
477
+ print(mix_ratio, mix_ratio.dtype, self.white.dtype, self.white.device, self.state['background'].latent.dtype, self.state['background'].latent.device)
478
  self.bootstrap_latent = mix_ratio * self.white + (1.0 - mix_ratio) * self.state['background'].latent
479
 
480
  self.ready_checklist['background_registered'] = True