Pass positive/negative prompt

bghira · Jul 7, 2023 · 5ba5cc9 · 5ba5cc9
1 parent a55b637
commit 5ba5cc9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/discord_tron_client/classes/image_manipulation/pipeline.py b/discord_tron_client/classes/image_manipulation/pipeline.py
@@ -217,6 +217,8 @@ def _run_pipeline(
                 # 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,