diff --git a/modelscope/utils/ast_utils.py b/modelscope/utils/ast_utils.py index 3775748d..657ebb33 100644 --- a/modelscope/utils/ast_utils.py +++ b/modelscope/utils/ast_utils.py @@ -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(