fix: add logger import

Signed-off-by: Markus Schuettler <markus.schuettler@tngtech.com>
TNG · Feb 28, 2025 · 5a65baf · 5a65baf
1 parent 46f5e7b
commit 5a65baf
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/service/comfyui_downloader.py b/service/comfyui_downloader.py
@@ -209,6 +209,7 @@ def download_custom_node(node_repo_data: ComfyUICustomNodesGithubRepoId) -> bool
 SCORE = 0.965 # 0.965 and less - is safety content
 
 logging.getLogger('transformers').setLevel(logging.ERROR)
+from scripts.reactor_logger import logger
 
 def nsfw_image(img_path: str, model_path: str):
     with Image.open(img_path) as img: