prompt emb

This commit is contained in:
Johannes Stelzer 2024-01-09 14:12:05 +01:00
parent 7c48f5e234
commit b6b921788c
1 changed files with 2 additions and 2 deletions

View File

@ -95,12 +95,12 @@ class DiffusersHolder():
def get_text_embedding(self, prompt): def get_text_embedding(self, prompt):
text_embeddings = self.pipe.encode_prompt( text_embeddings = self.pipe.encode_prompt(
prompt=prompt, prompt=prompt,
prompt_2=None, prompt_2=prompt,
device=self.pipe._execution_device, device=self.pipe._execution_device,
num_images_per_prompt=1, num_images_per_prompt=1,
do_classifier_free_guidance=self.pipe.do_classifier_free_guidance, do_classifier_free_guidance=self.pipe.do_classifier_free_guidance,
negative_prompt=self.negative_prompt, negative_prompt=self.negative_prompt,
negative_prompt_2=None, negative_prompt_2=self.negative_prompt,
prompt_embeds=None, prompt_embeds=None,
negative_prompt_embeds=None, negative_prompt_embeds=None,
pooled_prompt_embeds=None, pooled_prompt_embeds=None,