diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_pipeline_factory.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tests/test_pipeline_factory.py b/tests/test_pipeline_factory.py index 408133b..14b771b 100644 --- a/tests/test_pipeline_factory.py +++ b/tests/test_pipeline_factory.py @@ -91,11 +91,15 @@ def test_base_pipeline_requires_vae_and_applies_all_loras(app_config, monkeypatc external_vae = object() runtime = (SimpleNamespace(float16="float16"), fake_diffusers(pipeline, external_vae)) monkeypatch.setattr(factory, "_runtime", lambda: runtime) + monkeypatch.setattr( + PipelineFactory, "_vae_config_dir", staticmethod(lambda cache_dir: "fake_vae_config") + ) result = factory.create_base() assert result is pipeline assert pipeline.vae is external_vae + assert pipeline.vae_load["config"] == "fake_vae_config" # type: ignore[attr-defined] assert pipeline.adapters == (["configured_lora_0", "configured_lora_1"], [0.8, 0.5]) assert [item["weight_name"] for item in pipeline.lora_loads] == [ "detail.safetensors", |