fix

Haidra-Org · Sep 13, 2024 · f6bf397 · f6bf397
1 parent 0497891
commit f6bf397
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/horde/bridge_reference.py b/horde/bridge_reference.py
@@ -186,7 +186,7 @@ def parse_bridge_agent(bridge_agent):
 @logger.catch(reraise=True)
 def check_bridge_capability(capability, bridge_agent):
     bridge_name, bridge_version = parse_bridge_agent(bridge_agent)
-    logger.debug([bridge_name, bridge_version])
+    # logger.debug([bridge_name, bridge_version])
     if bridge_name not in BRIDGE_CAPABILITIES:
         return False
     total_capabilities = set()

diff --git a/horde/classes/stable/worker.py b/horde/classes/stable/worker.py
@@ -157,7 +157,7 @@ def can_generate(self, waiting_prompt):
         if not waiting_prompt.safe_ip and not self.allow_unsafe_ipaddr:
             return [False, "unsafe_ip"]
         if self.limit_max_steps:
-            for mn in waiting_prompt.model_names():
+            for mn in waiting_prompt.get_model_names():
                 avg_steps = (
                     int(
                         model_reference.get_model_requirements(mn).get("min_steps", 20)

diff --git a/tests/test_image.py b/tests/test_image.py
@@ -116,7 +116,7 @@ def test_flux_image_gen(api_key: str, HORDE_URL: str, CIVERSION: str) -> None:
             "sampler_name": "k_euler",
         },
         "models": TEST_MODELS_FLUX,
-        # "extra_slow_workers": True,
+        "extra_slow_workers": True,
     }
     protocol = "http"
     if HORDE_URL in ["dev.stablehorde.net", "stablehorde.net"]:
@@ -129,7 +129,7 @@ def test_flux_image_gen(api_key: str, HORDE_URL: str, CIVERSION: str) -> None:
     pop_dict = {
         "name": "CICD Fake Dreamer",
         "models": TEST_MODELS_FLUX,
-        "bridge_agent": "AI Horde Worker reGen:9.0.0-citests:https://github.com/Haidra-Org/horde-worker-reGen",
+        "bridge_agent": "AI Horde Worker reGen:9.1.0-citests:https://github.com/Haidra-Org/horde-worker-reGen",
         "nsfw": True,
         "amount": 10,
         "max_pixels": 4194304,
@@ -140,8 +140,8 @@ def test_flux_image_gen(api_key: str, HORDE_URL: str, CIVERSION: str) -> None:
         "allow_controlnet": True,
         "allow_sdxl_controlnet": True,
         "allow_lora": True,
-        # "extra_slow_worker": True,
-        # "limit_max_steps": True,
+        "extra_slow_worker": True,
+        "limit_max_steps": True,
     }
     pop_req = requests.post(f"{protocol}://{HORDE_URL}/api/v2/generate/pop", json=pop_dict, headers=headers)
     try: