Stable-X commited on
Commit
4bca7b1
1 Parent(s): ba7b123

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,