Mirror the upstream implementation

bghira · Jul 7, 2023 · 98154ec · 98154ec
1 parent 5ba5cc9
commit 98154ec
Showing 1 changed file with 2 additions and 10 deletions.
diff --git a/discord_tron_client/classes/image_manipulation/pipeline.py b/discord_tron_client/classes/image_manipulation/pipeline.py
@@ -216,14 +216,6 @@ def _run_pipeline(
             if not promptless_variation and image is None:
                 # text2img workflow
                 if "ptx0/s1" in user_config.get("model", ""):
-                    encoded_embeds = pipe.encode_prompt(
-                        prompt=positive_prompt,
-                        negative_prompt=negative_prompt,
-                        device=self.pipeline_manager.device,
-                        num_images_per_prompt=batch_size,
-                        prompt_embeds=prompt_embed,
-                        negative_prompt_embeds=negative_embed,
-                    )
                     preprocessed_images = pipe(
                         prompt_embeds=prompt_embed,
                         num_images_per_prompt=batch_size,
@@ -235,8 +227,8 @@ def _run_pipeline(
                         guidance_scale=guidance_scale,
                         output_type=image_return_type,
                         generator=generator,
-                        pooled_prompt_embeds=encoded_embeds[2],
-                        pooled_negative_prompt_embeds=encoded_embeds[3]
+                        pooled_prompt_embeds=prompt_embed[0],
+                        pooled_negative_prompt_embeds=negative_embed[0]
                     ).images
                 elif "ptx0/s2" in user_config.get("model", ""):
                     preprocessed_images = pipe(