From c23487408f913e160fceb8af43385efc699386e1 Mon Sep 17 00:00:00 2001 From: Zhicheng Zhang Date: Wed, 12 Apr 2023 20:24:11 +0800 Subject: [PATCH] bug fixed --- modelscope/trainers/builder.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/modelscope/trainers/builder.py b/modelscope/trainers/builder.py index 4dbaea56..8d1847f6 100644 --- a/modelscope/trainers/builder.py +++ b/modelscope/trainers/builder.py @@ -1,5 +1,6 @@ # Copyright (c) Alibaba, Inc. and its affiliates. from modelscope.metainfo import Trainers +from modelscope.pipelines.builder import normalize_model_input from modelscope.pipelines.util import is_official_hub_path from modelscope.utils.config import check_config from modelscope.utils.constant import DEFAULT_MODEL_REVISION @@ -31,8 +32,8 @@ def build_trainer(name: str = Trainers.default, default_args: dict = None): model, revision=model_revision) if isinstance( model, str) else read_config( model[0], revision=model_revision) - check_config(configuration) + model_dir = normalize_model_input(model, model_revision) register_plugins_repo(configuration.safe_get('plugins')) - register_modelhub_repo(model, + register_modelhub_repo(model_dir, configuration.get('allow_remote', False)) return build_from_cfg(cfg, TRAINERS, default_args=default_args)