[Logger] Now using get_debug_level and debugl from SeCoNoHe
This commit is contained in:
@@ -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")
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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):
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user