Test fix

pipeline.py

@@ -332,6 +332,7 @@ class Zero123PlusPipeline(diffusers.StableDiffusionPipeline):
         self,
         image: Image.Image = None,
         prompts = "",
+        prompt_embeds = None
         *args,
         num_images_per_prompt: Optional[int] = 1,
         guidance_scale=4.0,
@@ -365,14 +366,15 @@ class Zero123PlusPipeline(diffusers.StableDiffusionPipeline):
         global_embeds = encoded.image_embeds
         global_embeds = global_embeds.unsqueeze(-2)
         
-        encoder_hidden_states = self._encode_prompt(
-            prompts,
-            self.device,
-            num_images_per_prompt,
-            False
-        )
-        ramp = global_embeds.new_tensor(self.config.ramping_coefficients).unsqueeze(-1)
-        encoder_hidden_states = encoder_hidden_states + global_embeds * ramp
+        if prompt_embeds is not None
+            encoder_hidden_states = self._encode_prompt(
+                prompts,
+                self.device,
+                num_images_per_prompt,
+                False
+            )
+            ramp = global_embeds.new_tensor(self.config.ramping_coefficients).unsqueeze(-1)
+            encoder_hidden_states = encoder_hidden_states + global_embeds * ramp
         cak = dict(cond_lat=cond_lat)
         if hasattr(self.unet, "controlnet"):
             cak['control_depth'] = depth_image