ci: try not to hang up on lora download

Haidra-Org · Dec 31, 2023 · 0e03590 · 0e03590
1 parent a064615
commit 0e03590
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -108,6 +108,8 @@ def real_image() -> PIL.Image.Image:
 @pytest.fixture(scope="session")
 def lora_GlowingRunesAI(shared_model_manager: type[SharedModelManager]) -> str:
     assert shared_model_manager.manager.lora
+    if shared_model_manager.manager.lora.is_model_available("GlowingRunesAI"):
+        return "GlowingRunesAI"
     name = shared_model_manager.manager.lora.fetch_adhoc_lora("GlowingRunesAI")
     assert name is not None
     assert shared_model_manager.manager.lora.is_model_available(name)
@@ -119,6 +121,7 @@ def pytest_collection_modifyitems(items):
     """Modifies test items to ensure test modules run in a given order."""
     MODULES_TO_RUN_FIRST = [
         "tests.meta.test_packaging_errors",
+        "tests.test_internal_comfyui_failures",
         "tests.test_initialisation",
         "tests.test_cuda",
         "tests.test_utils",