Add SDXL formatter profiles
This commit is contained in:
+15
-2
@@ -4,12 +4,22 @@ try:
|
||||
from .caption_naturalizer import naturalize_caption
|
||||
from .caption_policy import caption_profile_choices
|
||||
from .krea_formatter import format_krea2_prompt
|
||||
from .sdxl_formatter import format_sdxl_prompt, sdxl_quality_preset_choices, sdxl_style_preset_choices
|
||||
from .sdxl_formatter import (
|
||||
format_sdxl_prompt,
|
||||
sdxl_formatter_profile_choices,
|
||||
sdxl_quality_preset_choices,
|
||||
sdxl_style_preset_choices,
|
||||
)
|
||||
except ImportError: # Allows local smoke tests from the repository root.
|
||||
from caption_naturalizer import naturalize_caption
|
||||
from caption_policy import caption_profile_choices
|
||||
from krea_formatter import format_krea2_prompt
|
||||
from sdxl_formatter import format_sdxl_prompt, sdxl_quality_preset_choices, sdxl_style_preset_choices
|
||||
from sdxl_formatter import (
|
||||
format_sdxl_prompt,
|
||||
sdxl_formatter_profile_choices,
|
||||
sdxl_quality_preset_choices,
|
||||
sdxl_style_preset_choices,
|
||||
)
|
||||
|
||||
|
||||
class SxCPCaptionNaturalizer:
|
||||
@@ -138,6 +148,7 @@ class SxCPSDXLFormatter:
|
||||
"source_text": ("STRING", {"default": "", "multiline": True}),
|
||||
"input_hint": (["auto", "metadata_json", "prompt"], {"default": "auto"}),
|
||||
"target": (["auto", "single", "softcore", "hardcore"], {"default": "auto"}),
|
||||
"formatter_profile": (sdxl_formatter_profile_choices(), {"default": "manual_controls"}),
|
||||
"style_preset": (sdxl_style_preset_choices(), {"default": "flat_vector_pony"}),
|
||||
"quality_preset": (sdxl_quality_preset_choices(), {"default": "pony_high"}),
|
||||
"trigger": ("STRING", {"default": "mythp0rt", "multiline": False}),
|
||||
@@ -174,6 +185,7 @@ class SxCPSDXLFormatter:
|
||||
source_text,
|
||||
input_hint,
|
||||
target,
|
||||
formatter_profile,
|
||||
style_preset,
|
||||
quality_preset,
|
||||
trigger,
|
||||
@@ -195,6 +207,7 @@ class SxCPSDXLFormatter:
|
||||
negative_prompt=negative_prompt or "",
|
||||
input_hint=input_hint,
|
||||
target=target,
|
||||
formatter_profile=formatter_profile,
|
||||
style_preset=style_preset,
|
||||
quality_preset=quality_preset,
|
||||
trigger=trigger,
|
||||
|
||||
Reference in New Issue
Block a user