diff options
Diffstat (limited to 'tests/test_config.py')
| -rw-r--r-- | tests/test_config.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tests/test_config.py b/tests/test_config.py index 5e4341a..c24064b 100644 --- a/tests/test_config.py +++ b/tests/test_config.py @@ -76,6 +76,8 @@ def test_loads_and_resolves_all_model_paths(tmp_path: Path) -> None: [ ('checkpoint = "../base.safetensors"', "file name"), ("steps = 0", "steps"), + ('scheduler = "dpmpp_2m"', "scheduler"), + ('scheduler = "euler"', "scheduler"), ('scheduler = "unknown"', "scheduler"), ('default_size = "1024"', "default_size"), ('preprocessor = "zoe"', "preprocessor"), @@ -120,3 +122,4 @@ def test_can_skip_path_checks_for_schema_only_use(tmp_path: Path) -> None: ) config = load_config(path, validate_paths=False) assert config.model.root == (tmp_path / "missing").resolve() + assert config.generation.scheduler == "dpmpp_2m_karras" |