From 4da4a7b49829b5fa96c58db9072ba9f8a35e3c8a Mon Sep 17 00:00:00 2001 From: "Salvador E. Tropea" Date: Wed, 23 Jul 2025 09:37:01 -0300 Subject: [PATCH] [Logger] Now using get_debug_level and debugl from SeCoNoHe --- src/nodes/db/hash_dir.py | 2 +- src/nodes/inference/get_model.py | 3 ++- src/nodes/utils/misc.py | 10 ---------- tool/demucs2safetensors.py | 3 ++- 4 files changed, 5 insertions(+), 13 deletions(-) diff --git a/src/nodes/db/hash_dir.py b/src/nodes/db/hash_dir.py index 3ad1d03..139b1e2 100644 --- a/src/nodes/db/hash_dir.py +++ b/src/nodes/db/hash_dir.py @@ -10,11 +10,11 @@ import os import csv import logging +from seconohe.logger import debugl import sys from .hash import get_hash from .. import NODES_NAME -from ..utils.misc import debugl # Set up the logger as specified logger = logging.getLogger(f"{NODES_NAME}.hash_dir") diff --git a/src/nodes/inference/get_model.py b/src/nodes/inference/get_model.py index 2fc1eda..1067bfd 100644 --- a/src/nodes/inference/get_model.py +++ b/src/nodes/inference/get_model.py @@ -8,9 +8,10 @@ import importlib import json import logging from safetensors import safe_open +from seconohe.logger import get_debug_level from .MDX_Net import MDX_Net from .. import NODES_NAME -from ..utils.misc import json_object_hook, get_debug_level +from ..utils.misc import json_object_hook # Demucs class imports from .demucs_api import BagOfModels from .demucs_log_helper import DemucsModelInfo diff --git a/src/nodes/utils/misc.py b/src/nodes/utils/misc.py index c96f4ec..1f00996 100644 --- a/src/nodes/utils/misc.py +++ b/src/nodes/utils/misc.py @@ -4,16 +4,6 @@ # Project: ComfyUI-AudioSeparation from fractions import Fraction import json -import logging - - -def get_debug_level(logger): - return logging.DEBUG - logger.getEffectiveLevel() + 1 - - -def debugl(logger, level, msg): - if get_debug_level(logger) >= level: - logger.debug(msg) def cli_add_verbose(parser): diff --git a/tool/demucs2safetensors.py b/tool/demucs2safetensors.py index 98f00de..6aa092c 100755 --- a/tool/demucs2safetensors.py +++ b/tool/demucs2safetensors.py @@ -14,6 +14,7 @@ import argparse import json from pathlib import Path from safetensors.torch import save_file +from seconohe.logger import debugl import sys import torch from typing import Dict @@ -26,7 +27,7 @@ try: except Exception: with_demuc_lib = False import bootstrap # noqa: F401 -from source.utils.misc import cli_add_verbose, FractionEncoder, debugl +from source.utils.misc import cli_add_verbose, FractionEncoder from source.utils.logger import main_logger, logger_set_standalone from source.db.models_db import cli_add_db, get_download_url, ModelsDB from source.db.hash import get_hash