From fd987cb31fd20483080f8d80ce35b53f101f243f Mon Sep 17 00:00:00 2001 From: tanaka takashi Date: Sat, 12 Sep 2026 20:12:17 +0900 Subject: feat: SD 1.x向けScheduler対応を5種類に整理 MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit generation.schedulerの正式対応値をdpmpp_2m_karras、dpmpp_2m_sde_karras、euler_a、ddim、lcmに限定し、既定値をdpmpp_2m_karrasへ変更する。旧dpmpp_2m、eulerおよび未知値は起動時の設定validationで拒否する。 Scheduler生成処理をPipelineFactoryへ集約する。DPM++ 2M系はDPMSolverMultistepSchedulerを共用し、algorithm_type、solver_order=2、use_karras_sigmas=Trueで通常版とSDE版を切り替える。Euler Ancestral、DDIM、LCMは現在のpipeline scheduler configを引き継いで構築する。 LCM-LoRAは既存のmodel.lora設定から通常のLoRAとして読み込み、LCM選択時にもstepsとcfg_scaleを自動調整せずTOML値をそのまま使用する。MCP Tool schemaにはscheduler入力を追加しない。 Specifications.md、Design.md、config.mdおよびconfig.example.tomlのenum、既定値、対応表、設定例を新仕様へ統一する。5 schedulerの変換、DPM++ constructor option、旧値と未知値の拒否、省略時の既定値、MCP schema非公開をテストで検証する。 検証: ruff check、ruff format --check、pyright、pytest（39 passed）、インストール済みDiffusersで5 schedulerの実インスタンス構築を確認。 --- src/mcp_sd15/config.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'src/mcp_sd15/config.py') 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" -- cgit v1.2.3