diff options
Diffstat (limited to 'src/mcp_sd15/config.py')
| -rw-r--r-- | src/mcp_sd15/config.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/mcp_sd15/config.py b/src/mcp_sd15/config.py index 5bbe225..1a5a71d 100644 --- a/src/mcp_sd15/config.py +++ b/src/mcp_sd15/config.py @@ -11,7 +11,7 @@ from typing import Any from .errors import ConfigError VALID_SIZES = frozenset({"square_512", "landscape_512", "portrait_512"}) -VALID_SCHEDULERS = frozenset({"dpmpp_2m", "ddim", "euler", "euler_a"}) +VALID_SCHEDULERS = frozenset({"dpmpp_2m_karras", "dpmpp_2m_sde_karras", "euler_a", "ddim", "lcm"}) @dataclass(frozen=True) @@ -41,7 +41,7 @@ class ModelConfig: class GenerationConfig: steps: int = 25 cfg_scale: float = 7.0 - scheduler: str = "dpmpp_2m" + scheduler: str = "dpmpp_2m_karras" default_size: str = "square_512" @@ -189,7 +189,7 @@ def load_config(path: str | Path, *, validate_paths: bool = True) -> AppConfig: cfg_scale = _number(generation_raw, "cfg_scale", 7.0) if not 0 <= cfg_scale <= 30: raise ConfigError("generation.cfg_scale must be between 0 and 30") - scheduler = _string(generation_raw, "scheduler") or "dpmpp_2m" + scheduler = _string(generation_raw, "scheduler") or "dpmpp_2m_karras" if scheduler not in VALID_SCHEDULERS: raise ConfigError(f"unsupported scheduler: {scheduler}") default_size = _string(generation_raw, "default_size") or "square_512" |