This commit is contained in:
Johannes Stelzer 2024-01-09 14:16:05 +01:00
parent e280494162
commit c155f37b04
1 changed files with 1 additions and 1 deletions

View File

@ -93,7 +93,7 @@ class DiffusersHolder():
self.negative_prompt = [self.negative_prompt[0]] self.negative_prompt = [self.negative_prompt[0]]
def get_text_embedding(self, prompt): def get_text_embedding(self, prompt):
do_classifier_free_guidance = self._guidance_scale > 1 and self.unet.config.time_cond_proj_dim is None do_classifier_free_guidance = self.guidance_scale > 1 and self.unet.config.time_cond_proj_dim is None
text_embeddings = self.pipe.encode_prompt( text_embeddings = self.pipe.encode_prompt(
prompt=prompt, prompt=prompt,
prompt_2=prompt, prompt_2=prompt,