test

Haidra-Org · Sep 13, 2024 · c2f6bc4 · c2f6bc4
1 parent b6dc735
commit c2f6bc4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/horde/classes/stable/worker.py b/horde/classes/stable/worker.py
@@ -140,6 +140,7 @@ def can_generate(self, waiting_prompt):
             and not check_bridge_capability("stable_cascade_2pass", self.bridge_agent)
         ):
             return [False, "bridge_version"]
+        logger.debug('test')
         if "flux_1" in model_reference.get_all_model_baselines(self.get_model_names()) and not check_bridge_capability(
             "flux", self.bridge_agent
         ):

diff --git a/tests/test_image.py b/tests/test_image.py
@@ -7,7 +7,6 @@
 import requests
 
 TEST_MODELS = ["Fustercluck", "AlbedoBase XL (SDXL)"]
-TEST_MODELS = ["Flux.1-Schnell fp8 (Compact)"]
 
 
 def test_simple_image_gen(api_key: str, HORDE_URL: str, CIVERSION: str) -> None:
@@ -185,5 +184,5 @@ def test_flux_image_gen(api_key: str, HORDE_URL: str, CIVERSION: str) -> None:
 
 if __name__ == "__main__":
     # "ci/cd#12285"
-    test_simple_image_gen("2bc5XkMeLAWiN9O5s7bhfg", "dev.stablehorde.net", "0.1.1")
-    # test_flux_image_gen("2bc5XkMeLAWiN9O5s7bhfg", "dev.stablehorde.net", "0.1.1")
+    # test_simple_image_gen("2bc5XkMeLAWiN9O5s7bhfg", "dev.stablehorde.net", "0.1.1")
+    test_flux_image_gen("2bc5XkMeLAWiN9O5s7bhfg", "dev.stablehorde.net", "0.1.1")