diff options
| author | tanaka takashi <isazaka@gmail.com> | 2026-09-13 22:02:44 +0900 |
|---|---|---|
| committer | tanaka takashi <isazaka@gmail.com> | 2026-09-13 22:02:44 +0900 |
| commit | 213e738140859dbcc12c2540ce40d6b6d2875aa7 (patch) | |
| tree | 93443693ffbfbfb8706e1eea2f857d677d62cb4a /src/mcp_sd15/pipeline_factory.py | |
| parent | de44625e3baae4d43ad096248b7096578b11e0b0 (diff) | |
ddim timestep_spacingをtrailingに強制するモードを追加
Diffstat (limited to 'src/mcp_sd15/pipeline_factory.py')
| -rw-r--r-- | src/mcp_sd15/pipeline_factory.py | 5 |
1 files changed, 5 insertions, 0 deletions
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, |