Spaces:
Runtime error
Runtime error
Update model.py
Browse files
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
|