mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-03-21 21:22:11 -03:00
19 lines
511 B
Python
19 lines
511 B
Python
import os
|
|
import importlib
|
|
from .metadata_hook import MetadataHook
|
|
from .metadata_registry import MetadataRegistry
|
|
|
|
def init():
|
|
# Install hooks to collect metadata during execution
|
|
MetadataHook.install()
|
|
|
|
# Initialize registry
|
|
registry = MetadataRegistry()
|
|
|
|
print("ComfyUI Metadata Collector initialized")
|
|
|
|
def get_metadata(prompt_id=None):
|
|
"""Helper function to get metadata from the registry"""
|
|
registry = MetadataRegistry()
|
|
return registry.get_metadata(prompt_id)
|