ddim timestep_spacingをtrailingに強制するモードを追加

1 files changed, 11 insertions, 0 deletions

diff --git a/tests/test_config.py b/tests/test_config.py
index 5d339f8..0e34086 100644
--- a/tests/test_config.py
+++ b/tests/test_config.py
@@ -282,6 +282,17 @@ def test_generation_defaults_clip_skip_and_pad_last_block(tmp_path: Path) -> Non
     assert config.generation.pad_last_block is False
 
 
+def test_generation_accepts_ddim_trailing_scheduler(tmp_path: Path) -> None:
+    path = tmp_path / "config.toml"
+    path.write_text(
+        f'[model]\nroot = "{tmp_path.as_posix()}"\ncheckpoint = "base.safetensors"\n'
+        '[generation]\nscheduler = "ddim_trailing"\n',
+        encoding="utf-8",
+    )
+    config = load_config(path, validate_paths=False)
+    assert config.generation.scheduler == "ddim_trailing"
+
+
 def test_generation_reads_clip_skip_and_pad_last_block(tmp_path: Path) -> None:
     path = tmp_path / "config.toml"
     path.write_text(