Wuvin commited on
Commit
6babd58
1 Parent(s): b77390c
custum_3d_diffusion/custum_pipeline/unifield_pipeline_img2mvimg.py CHANGED
@@ -214,6 +214,7 @@ class StableDiffusionImage2MVCustomPipeline(
214
  emb_image = image
215
 
216
  image_embeddings = self._encode_image(emb_image, device, num_images_per_prompt, do_classifier_free_guidance)
 
217
  cond_latents = self.encode_latents(image, image_embeddings.device, image_embeddings.dtype, height_cond, width_cond)
218
  cond_latents = torch.cat([torch.zeros_like(cond_latents), cond_latents]) if do_classifier_free_guidance else cond_latents
219
  image_pixels = self.feature_extractor(images=emb_image, return_tensors="pt").pixel_values
 
214
  emb_image = image
215
 
216
  image_embeddings = self._encode_image(emb_image, device, num_images_per_prompt, do_classifier_free_guidance)
217
+ print("DEBUG: image_embeddings", image_embeddings.dtype, image_embeddings.device)
218
  cond_latents = self.encode_latents(image, image_embeddings.device, image_embeddings.dtype, height_cond, width_cond)
219
  cond_latents = torch.cat([torch.zeros_like(cond_latents), cond_latents]) if do_classifier_free_guidance else cond_latents
220
  image_pixels = self.feature_extractor(images=emb_image, return_tensors="pt").pixel_values