Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-16 16:27:45 +01:00
Code Issues Packages Projects Releases Wiki Activity

bug fixed

Browse Source
This commit is contained in:
Zhicheng Zhang
2023-04-12 16:47:23 +08:00
parent d81f32c5c9
commit 952d34f63c
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
modelscope/trainers/builder.py
View File

@@ -27,11 +27,12 @@ def build_trainer(name: str = Trainers.default, default_args: dict = None):
or (isinstance(model, list) and isinstance(model[0], str)):
if is_official_hub_path(model, revision=model_revision):
# read config file from hub and parse
cfg = read_config(
configuration = read_config(
model, revision=model_revision) if isinstance(
model, str) else read_config(
model[0], revision=model_revision)
check_config(cfg)
register_plugins_repo(cfg.safe_get('plugins'))
register_modelhub_repo(model, cfg.get('allow_remote', False))
check_config(configuration)
register_plugins_repo(configuration.safe_get('plugins'))
register_modelhub_repo(model,
configuration.get('allow_remote', False))
return build_from_cfg(cfg, TRAINERS, default_args=default_args)