From 213e738140859dbcc12c2540ce40d6b6d2875aa7 Mon Sep 17 00:00:00 2001
From: tanaka takashi <isazaka@gmail.com>
Date: Sun, 13 Sep 2026 22:02:44 +0900
Subject: ddim timestep_spacingをtrailingに強制するモードを追加
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/mcp_sd15/pipeline_factory.py | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'src/mcp_sd15/pipeline_factory.py')

diff --git a/src/mcp_sd15/pipeline_factory.py b/src/mcp_sd15/pipeline_factory.py
index 02fe882..e865555 100644
--- a/src/mcp_sd15/pipeline_factory.py
+++ b/src/mcp_sd15/pipeline_factory.py
@@ -151,6 +151,11 @@ class PipelineFactory:
                 solver_order=2,
                 use_karras_sigmas=True,
             )
+        if name == "ddim_trailing":
+            return diffusers.DDIMScheduler.from_config(
+                current.config,
+                timestep_spacing="trailing",
+            )
         classes = {
             "euler_a": diffusers.EulerAncestralDiscreteScheduler,
             "ddim": diffusers.DDIMScheduler,
-- 
cgit v1.2.3