refactor(logging): replace print statements with logger in metadata_collector

This commit is contained in:
Will Miao
2026-02-01 15:41:41 +08:00
parent 553e1868e1
commit db1aec94e5
2 changed files with 17 additions and 11 deletions

View File

@@ -1,4 +1,7 @@
import os
import logging
logger = logging.getLogger(__name__)
# Check if running in standalone mode
standalone_mode = os.environ.get("LORA_MANAGER_STANDALONE", "0") == "1" or os.environ.get("HF_HUB_DISABLE_TELEMETRY", "0") == "0"
@@ -14,7 +17,7 @@ if not standalone_mode:
# Initialize registry
registry = MetadataRegistry()
print("ComfyUI Metadata Collector initialized")
logger.info("ComfyUI Metadata Collector initialized")
def get_metadata(prompt_id=None):
"""Helper function to get metadata from the registry"""
@@ -23,7 +26,7 @@ if not standalone_mode:
else:
# Standalone mode - provide dummy implementations
def init():
print("ComfyUI Metadata Collector disabled in standalone mode")
logger.info("ComfyUI Metadata Collector disabled in standalone mode")
def get_metadata(prompt_id=None):
"""Dummy implementation for standalone mode"""