Spaces:
Running
on
Zero
Running
on
Zero
Update stablediffuse/pipeline_yoso_diffuse.py
Browse files
stablediffuse/pipeline_yoso_diffuse.py
CHANGED
@@ -587,7 +587,7 @@ class YOSODiffusePipeline(StableDiffusionControlNetPipeline):
|
|
587 |
image_latent = image_latent * self.vae.config.scaling_factor
|
588 |
image_latent = image_latent.repeat_interleave(ensemble_size, dim=0) # [N*E,4,h,w]
|
589 |
|
590 |
-
pred_latent = latents
|
591 |
if pred_latent is None:
|
592 |
pred_latent = randn_tensor(
|
593 |
image_latent.shape,
|
|
|
587 |
image_latent = image_latent * self.vae.config.scaling_factor
|
588 |
image_latent = image_latent.repeat_interleave(ensemble_size, dim=0) # [N*E,4,h,w]
|
589 |
|
590 |
+
pred_latent = torch.zeros_like(latents)
|
591 |
if pred_latent is None:
|
592 |
pred_latent = randn_tensor(
|
593 |
image_latent.shape,
|