diff --git a/combo/utils/download.py b/combo/utils/download.py index 7a928d2f81387d91945f6a82fa2d058ec195154a..e88c759f7b21e8feef57f5fb73d37f58acc192e6 100644 --- a/combo/utils/download.py +++ b/combo/utils/download.py @@ -20,7 +20,8 @@ _CACHE_DIR = os.getenv("COMBO_DIR", os.path.join(_HOME_DIR, ".combo")) def download_file(model_name, force=False): _make_cache_dir() - data = model_name.split("-")[-1] + split_name = model_name.split("-") + model_name, data = "-".join(split_name[:-1]), split_name[-1] url = _URL.format(model=model_name, data=DATA_TO_PATH[data]) local_filename = url.split("/")[-1] location = os.path.join(_CACHE_DIR, local_filename)