diff --git a/requirements.txt b/requirements.txt index 03796c46..7afadf58 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ # Add this in for tox, comment out for build --extra-index-url https://download.pytorch.org/whl/cu118 -horde_model_reference~=0.4.7 +horde_model_reference~=0.4.8 pydantic torch>=2.0.0 xformers>=0.0.19 diff --git a/tests/conftest.py b/tests/conftest.py index 606b3ae9..d85ee098 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -55,6 +55,9 @@ def shared_model_manager(hordelib_instance: HordeLib) -> Generator[type[SharedMo assert SharedModelManager.manager.blip is not None assert SharedModelManager.manager.clip is not None + # FIXME + SharedModelManager.manager.compvis.download_model("SDXL 1.0") + yield SharedModelManager SharedModelManager._instance = None # type: ignore diff --git a/tox.ini b/tox.ini index 5c19f0ef..86ac641e 100644 --- a/tox.ini +++ b/tox.ini @@ -119,6 +119,8 @@ passenv = HORDELIB_TESTING AIWORKER_CACHE_HOME TESTS_ONGOING +setenv = + HORDE_MODEL_REFERENCE_GITHUB_BRANCH = "SDXL" deps = pytest>=7 pytest-sugar