Swapped to using built-in safetensors header loader.

- Build-in loader avoids worst-case slow load using max length.

__init__.py

@@ -18,6 +18,7 @@ import json
 import requests
 requests.packages.urllib3.disable_warnings()
 
+import comfy.utils
 import folder_paths
 
 comfyui_model_uri = folder_paths.models_dir
@@ -142,11 +143,9 @@ def search_path_to_system_path(model_path):
 
 def get_safetensor_header(path):
     try:
-        with open(path, "rb") as f:
+        header_bytes = comfy.utils.safetensors_header(path)
-            length_of_header = struct.unpack("<Q", f.read(8))[0]
+        header_json = json.loads(header_bytes)
-            header_bytes = f.read(length_of_header)
+        return header_json if header_json is not None else {}
-            header_json = json.loads(header_bytes)
-            return header_json
     except:
         return {}