eliphatfs commited on
Commit
5006f82
1 Parent(s): bdbe43e

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +2 -0
pipeline.py CHANGED
@@ -343,9 +343,11 @@ class Zero123PlusPipeline(diffusers.StableDiffusionPipeline):
343
  if image is None:
344
  raise ValueError("Inputting embeddings not supported for this pipeline. Please pass an image.")
345
  assert not isinstance(image, torch.Tensor)
 
346
  image_1 = self.feature_extractor_vae(images=image, return_tensors="pt").pixel_values
347
  image_2 = self.feature_extractor_clip(images=image, return_tensors="pt").pixel_values
348
  if depth_image is not None and hasattr(self.unet, "controlnet"):
 
349
  depth_image = self.depth_transforms_multi(depth_image).to(
350
  device=self.unet.controlnet.device, dtype=self.unet.controlnet.dtype
351
  )
 
343
  if image is None:
344
  raise ValueError("Inputting embeddings not supported for this pipeline. Please pass an image.")
345
  assert not isinstance(image, torch.Tensor)
346
+ image = to_rgb_image(image)
347
  image_1 = self.feature_extractor_vae(images=image, return_tensors="pt").pixel_values
348
  image_2 = self.feature_extractor_clip(images=image, return_tensors="pt").pixel_values
349
  if depth_image is not None and hasattr(self.unet, "controlnet"):
350
+ depth_image = to_rgb_image(depth_image)
351
  depth_image = self.depth_transforms_multi(depth_image).to(
352
  device=self.unet.controlnet.device, dtype=self.unet.controlnet.dtype
353
  )