fix logging

This commit is contained in:
mulin.lyh
2024-07-03 19:18:49 +08:00
parent 696da5c272
commit 2a72d9b178

View File

@@ -21,7 +21,6 @@ from modelscope.metainfo import (CustomDatasets, Heads, Hooks, LR_Schedulers,
Preprocessors, TaskModels, Trainers)
from modelscope.utils.constant import Fields, Tasks
from modelscope.utils.file_utils import get_modelscope_cache_dir
from modelscope.utils.logger import get_logger
from modelscope.utils.registry import default_group
p = Path(__file__)
@@ -58,10 +57,8 @@ TEMPLATE_FILE = 'ast_index_file.py'
def get_ast_logger():
_logger = get_logger()
ast_logger = logging.getLogger('modelscope.ast')
for handler in _logger.handlers:
ast_logger.addHandler(handler)
ast_logger.setLevel(logging.INFO)
return ast_logger
@@ -722,7 +719,6 @@ def load_index(
cache_dir = os.getenv('MODELSCOPE_CACHE', indexer_file_dir)
index_file = os.getenv('MODELSCOPE_INDEX_FILE', indexer_file)
file_path = os.path.join(cache_dir, index_file)
logger.info(f'Loading ast index from {file_path}')
index = None
if force_rebuild:
@@ -732,6 +728,7 @@ def load_index(
# when developing, we need to generator as need.
if __is_develop_model():
logger.info(f'Loading ast index from {file_path}')
if os.path.exists(file_path): # already exist, check it's latest
wrapped_index = _load_index(file_path)
md5, files_mtime = file_scanner.files_mtime_md5(