Disty0
/

zero123plus-pipeline

Model card Files Files and versions

Disty0 commited on Oct 25, 2023

Commit

4bbfb31

·

1 Parent(s): 56520d0

Test

Files changed (1) hide show

pipeline.py +2 -2

pipeline.py CHANGED Viewed

@@ -331,7 +331,7 @@ class Zero123PlusPipeline(diffusers.StableDiffusionPipeline):
     def __call__(
         self,
         image: Image.Image = None,
-        prompts = "",
         prompt_embeds = None,
         *args,
         num_images_per_prompt: Optional[int] = 1,
@@ -368,7 +368,7 @@ class Zero123PlusPipeline(diffusers.StableDiffusionPipeline):
         if prompt_embeds is None:
             encoder_hidden_states = self._encode_prompt(
-                prompts,
                 self.device,
                 num_images_per_prompt,
                 False

     def __call__(
         self,
         image: Image.Image = None,
+        prompt = "image",
         prompt_embeds = None,
         *args,
         num_images_per_prompt: Optional[int] = 1,
         if prompt_embeds is None:
             encoder_hidden_states = self._encode_prompt(
+                prompt,
                 self.device,
                 num_images_per_prompt,
                 False