Use pooled embeds

bghira · Jul 7, 2023 · a55b637 · a55b637
1 parent 95150d1
commit a55b637
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/discord_tron_client/classes/image_manipulation/pipeline.py b/discord_tron_client/classes/image_manipulation/pipeline.py
@@ -233,7 +233,7 @@ def _run_pipeline(
                         guidance_scale=guidance_scale,
                         output_type=image_return_type,
                         generator=generator,
-                        pooled_prompt_embeds=encoded_embeds[2]
+                        pooled_prompt_embeds=encoded_embeds[2],
                         pooled_negative_prompt_embeds=encoded_embeds[3]
                     ).images
                 elif "ptx0/s2" in user_config.get("model", ""):