Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-04 03:50:44 +02:00
Code Issues Packages Projects Releases Wiki Activity

Small fix for refiner (untested).

Browse Source
This commit is contained in:
Hameer Abbasi
2023-08-06 07:24:20 +00:00
parent bb66b5ce0e
commit 00a5df5806
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/prompt_parser_diffusers.py
+5 -2
View File
@@ -26,7 +26,7 @@ def compel_encode_prompt(pipeline: typing.Any, *args, **kwargs):
raise TypeError(f"Compel encoding not yet supported for {type(pipeline).__name__}.")
return compel_encode_fn(pipeline, *args, **kwargs)
def compel_encode_prompt_sdxl(pipeline: diffusers.StableDiffusionXLPipeline, prompt: str, negative_prompt: str, prompt_2: typing.Optional[str]=None, negative_prompt_2: typing.Optional[str]=None, is_refiner: bool = False):
def compel_encode_prompt_sdxl(pipeline: diffusers.StableDiffusionXLPipeline, prompt: str, negative_prompt: str, prompt_2: typing.Optional[str]=None, negative_prompt_2: typing.Optional[str]=None, is_refiner: bool = None):
if shared.opts.data['prompt_attention'] != 'Compel parser':
prompt = convert_to_compel(prompt)
negative_prompt = convert_to_compel(negative_prompt)
@@ -62,4 +62,7 @@ def compel_encode_prompt_sdxl(pipeline: diffusers.StableDiffusionXLPipeline, pro
[prompt_embed, negative_embed] = compel_te2.pad_conditioning_tensors_to_same_length([positive, negative])
return prompt_embed, positive_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,
diffusers.StableDiffusionImg2ImgPipeline: compel_encode_prompt_sdxl,
}