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

feat: Enhance checkpoint processing in format_metadata to handle non-string types safely

Browse Source
This commit is contained in:
Will Miao
2025-04-17 09:38:20 +08:00
parent 32d34d1748
commit 5fd069d70d
1 changed files with 14 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
py/nodes/save_image.py
View File

@@ -159,11 +159,20 @@ class SaveImage:
# Model info
if 'checkpoint' in metadata_dict:
# Extract basename without path
checkpoint = os.path.basename(metadata_dict.get('checkpoint', ''))
# Remove extension if present
checkpoint = os.path.splitext(checkpoint)[0]
params.append(f"Model: {checkpoint}")
# Ensure checkpoint is a string before processing
checkpoint = metadata_dict.get('checkpoint')
if checkpoint is not None:
# Handle both string and other types safely
if isinstance(checkpoint, str):
# Extract basename without path
checkpoint = os.path.basename(checkpoint)
# Remove extension if present
checkpoint = os.path.splitext(checkpoint)[0]
else:
# Convert non-string to string
checkpoint = str(checkpoint)
params.append(f"Model: {checkpoint}")
# Add LoRA hashes if available
if lora_hashes: