Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-08-29 16:41:01 +02:00
Code Issues Packages Projects Releases Wiki Activity

Check based on sd_model_type.

Browse Source
This commit is contained in:
Hameer Abbasi
2023-08-06 11:22:33 +00:00
parent 764c15aa9f
commit 7ea4f32cec
1 changed files with 6 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/prompt_parser_diffusers.py
+6 -17
View File
@@ -26,17 +26,7 @@ def convert_to_compel(prompt: str):
return converted_prompt
def compel_encode_prompt(pipeline: typing.Any, *args, **kwargs):
compel_encode_fn = COMPEL_ENCODE_FN_DICT.get(type(pipeline), None)
if compel_encode_fn is None:
shared.log.warning(
f"Compel encoding not yet supported for {type(pipeline).__name__}."
)
return (None,) * 4
return compel_encode_fn(pipeline, *args, **kwargs)
def compel_encode_prompt_sdxl(
def compel_encode_prompt(
pipeline: diffusers.StableDiffusionXLPipeline,
prompt: str,
negative_prompt: str,
@@ -44,6 +34,11 @@ def compel_encode_prompt_sdxl(
negative_prompt_2: typing.Optional[str] = None,
is_refiner: bool = None,
):
if shared.sd_model_type not in {"sd", "sdxl"}:
shared.log.warning(
f"Compel encoding not yet supported for {type(pipeline).__name__}."
)
return (None,) * 4
if shared.opts.data["prompt_attention"] != "Compel parser":
prompt = convert_to_compel(prompt)
negative_prompt = convert_to_compel(negative_prompt)
@@ -80,9 +75,3 @@ def compel_encode_prompt_sdxl(
[positive, negative]
)
return prompt_embed, positive_pooled, negative_embed, negative_pooled
COMPEL_ENCODE_FN_DICT = {
diffusers.StableDiffusionXLPipeline: compel_encode_prompt_sdxl,
diffusers.StableDiffusionImg2ImgPipeline: compel_encode_prompt_sdxl,
}