Linting fixes.

This commit is contained in:
root
2023-08-04 15:16:13 +00:00
parent fdbafbd713
commit 076acf0664
6 changed files with 9 additions and 9 deletions
+3 -3
View File
@@ -35,9 +35,9 @@ def compel_encode_prompt_sdxl(pipeline: diffusers.StableDiffusionXLPipeline, pro
else:
positive, pooled = compel_te2(prompt)
negative, negative_pooled = compel_te2(negative_prompt)
[prompt_embed, negative_embed] = compel_te2.pad_conditioning_tensors_to_same_length([positive, negative])
return prompt_embed, pooled, negative_embed, negative_pooled
COMPEL_ENCODE_FN_DICT = {diffusers.StableDiffusionXLPipeline: compel_encode_prompt_sdxl}
COMPEL_ENCODE_FN_DICT = {diffusers.StableDiffusionXLPipeline: compel_encode_prompt_sdxl}