feat: add default download model configuration

This commit is contained in:
SWHL committed 2026-06-09 23:04:08 +08:00
1 parent e8938d8c2a
commit a4c7382bba
2 files changed
+9 -2

No files matched your search

+1 -1
View File
@@ -412,7 +412,7 @@ def parse_args(arg_list: Optional[List[str]] = None):
parser_check.set_defaults(func=check_install)
parser_download = subparser.add_parser("download_models", help="Download models")
parser_download.add_argument("--config", type=str)
parser_download.add_argument("--config", type=str, default=None)
parser_download.set_defaults(func=download_models)
args = parser.parse_args(arg_list)
+8 -1
View File
@@ -14,7 +14,14 @@ from .utils import mkdir
root_dir = Path(__file__).resolve().parent.parent
def download_models(config_path: Union[str, Path]) -> None:
def download_models(config_path: Union[str, Path, None] = None) -> None:
if config_path is None:
config_path = root_dir / "config.yaml"
else:
config_path = Path(config_path).expanduser().resolve()
if not config_path.is_file():
raise FileNotFoundError(f"Config file not found: {config_path}")
cfg = ParseParams.load(config_path)
model_root_dir = cfg.Global.get("model_root_dir")