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,