Merge pull request #227 from Haidra-Org/main

Haidra-Org · Mar 23, 2024 · 0b1120b · 0b1120b
2 parents 100197f + ed9d1c7
commit 0b1120b
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/hordelib/consts.py b/hordelib/consts.py
@@ -6,7 +6,7 @@
 
 from hordelib.config_path import get_hordelib_path
 
-COMFYUI_VERSION = "0c55f16c9e66eaa4915e288b34e4f848fb2d949f"
+COMFYUI_VERSION = "a28a9dc83684624ee2167c0b92d976bb68f2c606"
 """The exact version of ComfyUI version to load."""
 
 REMOTE_PROXY = ""
@@ -52,7 +52,7 @@ class MODEL_CATEGORY_NAMES(StrEnum):
     # MODEL_CATEGORY_NAMES.diffusers: True,
     MODEL_CATEGORY_NAMES.esrgan: True,
     MODEL_CATEGORY_NAMES.gfpgan: True,
-    MODEL_CATEGORY_NAMES.safety_checker: True,
+    # MODEL_CATEGORY_NAMES.safety_checker: True,
     MODEL_CATEGORY_NAMES.lora: True,
     MODEL_CATEGORY_NAMES.ti: True,
 }

diff --git a/hordelib/model_manager/ti.py b/hordelib/model_manager/ti.py
@@ -310,13 +310,20 @@ def _download_thread(self, thread_number):
                                 "Fewer retries will be attempted.",
                             )
 
+                        hordeling_json = hordeling_response.json()
                         # We will retry
                         logger.debug(
                             "AI Hordeling reported error when downloading metadata "
                             f"for Textual Inversion: {ti['filename']}: "
-                            f"{hordeling_response.json()}"
+                            f"{hordeling_json}"
                             f"Retry {retries}/{self.MAX_RETRIES}",
                         )
+
+                        message = hordeling_json.get("message", "")
+                        if message is not None and isinstance(message, str) and "hash" in message.lower():
+                            logger.debug(f"Textual Inversion: {ti['filename']} hash mismatch reported.")
+                            break
+
                     else:
                         hordeling_json = hordeling_response.json()
                         if hordeling_json.get("sha256"):