From e6e7849c31d8f81b1490e0d94fe17f7f5017d4c5 Mon Sep 17 00:00:00 2001 From: tazlin Date: Mon, 19 Feb 2024 10:53:06 -0500 Subject: [PATCH] tests: download/validate `AlbedoBase XL (SDXL)` for ci This alleviates test failures on fresh CI machines --- tests/conftest.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/conftest.py b/tests/conftest.py index cd3cbfe4..d69995ea 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -66,6 +66,8 @@ def shared_model_manager(hordelib_instance: HordeLib) -> Generator[type[SharedMo assert SharedModelManager.manager.validate_model("Deliberate") assert SharedModelManager.manager.download_model("SDXL 1.0") assert SharedModelManager.manager.validate_model("SDXL 1.0") + assert SharedModelManager.manager.download_model("AlbedoBase XL (SDXL)") + assert SharedModelManager.manager.validate_model("AlbedoBase XL (SDXL)") assert SharedModelManager.manager.controlnet is not None assert SharedModelManager.manager.controlnet.download_all_models()