fixed: compatibility patch

EvilBT/ComfyUI_SLK_joy_caption_two#109
ltdrdata · Jan 2, 2025 · 4401965 · 4401965
1 parent d22e2fa
commit 4401965
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/__init__.py b/__init__.py
@@ -8,7 +8,7 @@
 import importlib
 import logging
 
-version_code = [1, 2, 2]
+version_code = [1, 2, 3]
 version_str = f"V{version_code[0]}.{version_code[1]}" + (f'.{version_code[2]}' if len(version_code) > 2 else '')
 logging.info(f"### Loading: ComfyUI-Impact-Subpack ({version_str})")
 

diff --git a/modules/subcore.py b/modules/subcore.py
@@ -136,7 +136,8 @@ def torch_wrapper(*args, **kwargs):
         return orig_torch_load(*args, **kwargs)
     else:
         logging.warning("[Impact Subpack] Your torch version is outdated, and security features cannot be applied properly.")
-        return orig_torch_load(*args, **kwargs, weights_only=False)
+        kwargs['weights_only'] = False
+        return orig_torch_load(*args, **kwargs)
 
 torch.load = torch_wrapper
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 name = "comfyui-impact-subpack"
 description = "This node pack provides nodes that complement the Impact Pack, such as the UltralyticsDetectorProvider."
-version = "1.2.2"
+version = "1.2.3"
 license = { file = "LICENSE.txt" }
 dependencies = ["matplotlib", "ultralytics==8.3.40"]