Skip to content

Commit 9bdb018

Browse files
committed
update modelzoo utils: list available models from huggingface
1 parent 07c391d commit 9bdb018

1 file changed

Lines changed: 2 additions & 4 deletions

File tree

dlclive/modelzoo/utils.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
from pathlib import Path
1010

1111
from dlclibrary.dlcmodelzoo.modelzoo_download import download_huggingface_model
12+
from dlclibrary.dlcmodelzoo.modelzoo_download import _load_model_names as huggingface_model_paths
1213
from ruamel.yaml import YAML
1314

1415
from dlclive.modelzoo.resolve_config import update_config
@@ -49,10 +50,7 @@ def list_available_projects() -> list[str]:
4950

5051

5152
def list_available_combinations() -> list[str]:
52-
models = list_available_models()
53-
projects = list_available_projects()
54-
combinations = ["_".join([p, m]) for p in projects for m in models]
55-
return combinations
53+
return list(huggingface_model_paths.keys())
5654

5755

5856
def read_config_as_dict(config_path: str | Path) -> dict:

0 commit comments

Comments
 (0)