diff --git a/demucs/api.py b/demucs/api.py index deb79cdd..e61670f6 100644 --- a/demucs/api.py +++ b/demucs/api.py @@ -369,7 +369,7 @@ def save_audio( raise ValueError(f"Invalid suffix for path: {suffix}") -def list_models(repo: Optional[Path] = None) -> Dict[str, List[str]]: +def list_models(repo: Optional[Path] = None, remote_list: Optional[Path] = None) -> Dict[str, List[str]]: """ List the available models. Please remember that not all the returned models can be successfully loaded. @@ -385,9 +385,11 @@ def list_models(repo: Optional[Path] = None) -> Dict[str, List[str]]: """ model_repo: ModelOnlyRepo if repo is None: - models = _parse_remote_files(REMOTE_ROOT / 'files.txt') + if remote_list is None: + remote_list = REMOTE_ROOT + models = _parse_remote_files(remote_list / 'files.txt') model_repo = RemoteRepo(models) - bag_repo = BagOnlyRepo(REMOTE_ROOT, model_repo) + bag_repo = BagOnlyRepo(remote_list, model_repo) else: if not repo.is_dir(): fatal(f"{repo} must exist and be a directory.")