From b6b921788ca4b9ddacab0f75f5a90cd993a9e249 Mon Sep 17 00:00:00 2001
From: Johannes Stelzer <jsdmail@gmail.com>
Date: Tue, 9 Jan 2024 14:12:05 +0100
Subject: [PATCH] prompt emb

---
 diffusers_holder.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/diffusers_holder.py b/diffusers_holder.py
index 71f4d41..bcdf46e 100644
--- a/diffusers_holder.py
+++ b/diffusers_holder.py
@@ -95,12 +95,12 @@ class DiffusersHolder():
     def get_text_embedding(self, prompt):
         text_embeddings = self.pipe.encode_prompt(
             prompt=prompt,
-            prompt_2=None,
+            prompt_2=prompt,
             device=self.pipe._execution_device,
             num_images_per_prompt=1,
             do_classifier_free_guidance=self.pipe.do_classifier_free_guidance,
             negative_prompt=self.negative_prompt,
-            negative_prompt_2=None,
+            negative_prompt_2=self.negative_prompt,
             prompt_embeds=None,
             negative_prompt_embeds=None,
             pooled_prompt_embeds=None,