diff --git a/combo/utils/download.py b/combo/utils/download.py index e88c759f7b21e8feef57f5fb73d37f58acc192e6..f5deb82aa4405d849a806e7b972734bf1846746a 100644 --- a/combo/utils/download.py +++ b/combo/utils/download.py @@ -21,7 +21,7 @@ _CACHE_DIR = os.getenv("COMBO_DIR", os.path.join(_HOME_DIR, ".combo")) def download_file(model_name, force=False): _make_cache_dir() split_name = model_name.split("-") - model_name, data = "-".join(split_name[:-1]), split_name[-1] + data = 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)