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

feat: Enhance image retrieval in MetadataRegistry and update recipe routes to process images from metadata

Browse Source
This commit is contained in:
Will Miao
2025-04-18 09:24:48 +08:00
parent df6d56ce66
commit 91b4827c1d
2 changed files with 68 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
py/metadata_collector/metadata_registry.py
View File

@@ -239,6 +239,14 @@ class MetadataRegistry:
metadata = self.prompt_metadata[key]
if IMAGES in metadata and "first_decode" in metadata[IMAGES]:
return metadata[IMAGES]["first_decode"]["image"]
image_data = metadata[IMAGES]["first_decode"]["image"]
# If it's an image batch or tuple, handle various formats
if isinstance(image_data, (list, tuple)) and len(image_data) > 0:
# Return first element of list/tuple
return image_data[0]
# If it's a tensor, return as is for processing in the route handler
return image_data
return None