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

View File

@ -93,7 +93,7 @@ class DiffusersHolder():
self.negative_prompt = [self.negative_prompt[0]]
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.pipe.unet.config.time_cond_proj_dim is None
text_embeddings = self.pipe.encode_prompt(
prompt=prompt,
prompt_2=prompt,