XXX: fix: download SDXL

Haidra-Org · Aug 27, 2023 · d37a7cc · d37a7cc
1 parent 8554eee
commit d37a7cc
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git 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
@@ -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
@@ -119,6 +119,8 @@ passenv =
     HORDELIB_TESTING
     AIWORKER_CACHE_HOME
     TESTS_ONGOING
+setenv = 
+    HORDE_MODEL_REFERENCE_GITHUB_BRANCH = "SDXL"
 deps =
     pytest>=7
     pytest-sugar