alphard/ComfyUI-Lora-Manager
1
0
Fork 0
mirror of https://github.com/willmiao/ComfyUI-Lora-Manager.git synced 2026-06-09 12:39:23 -03:00
Code Issues Packages Projects Releases Wiki Activity

fix(utils): catch MemoryError in read_safetensors_metadata for non-safetensors files

Browse Source
This commit is contained in:
Will Miao
2026-06-05 16:30:50 +08:00
parent 01dac57c35
commit d9ee9b3155
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
py/utils/file_utils.py
View File

@@ -81,7 +81,7 @@ def read_safetensors_metadata(file_path: str) -> dict[str, Any]:
return {}
header = json.loads(header_bytes.decode("utf-8"))
return header.get("__metadata__", {})
except (OSError, json.JSONDecodeError, UnicodeDecodeError, struct.error):
except (OSError, json.JSONDecodeError, UnicodeDecodeError, struct.error, MemoryError, Exception):
return {}