alphard/ComfyUI-Lora-Manager
1
0
Fork 0
mirror of https://github.com/willmiao/ComfyUI-Lora-Manager.git synced 2026-03-21 21:22:11 -03:00
Code Issues Packages Projects Releases Wiki Activity

refactor: enhance hash processing in AutomaticMetadataParser for improved key handling

Browse Source
This commit is contained in:
Will Miao
2025-05-07 05:29:16 +08:00
parent bfd4eb3e11
commit 1c6e9d0b69
1 changed files with 22 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
py/recipes/parsers/automatic.py
View File

@@ -84,7 +84,28 @@ class AutomaticMetadataParser(RecipeMetadataParser):
if hashes_match:
try:
hashes = json.loads(hashes_match.group(1))
metadata["hashes"] = hashes
# Process hash keys
processed_hashes = {}
for key, value in hashes.items():
# Convert Model: or LORA: prefix to lowercase if present
if ':' in key:
prefix, name = key.split(':', 1)
prefix = prefix.lower()
else:
prefix = ''
name = key
# Clean up the name part
if '/' in name:
name = name.split('/')[-1] # Get last part after /
if '.safetensors' in name:
name = name.split('.safetensors')[0] # Remove .safetensors
# Reconstruct the key
new_key = f"{prefix}:{name}" if prefix else name
processed_hashes[new_key] = value
metadata["hashes"] = processed_hashes
# Remove hashes from params section to not interfere with other parsing
params_section = params_section.replace(hashes_match.group(0), '')
except json.JSONDecodeError: