diff --git a/combo/utils/download.py b/combo/utils/download.py index f5deb82aa4405d849a806e7b972734bf1846746a..448a9941d82b8b2b144ecd3301ea15f0886bba7c 100644 --- a/combo/utils/download.py +++ b/combo/utils/download.py @@ -22,7 +22,7 @@ def download_file(model_name, force=False): _make_cache_dir() split_name = model_name.split("-") data = split_name[-1] - url = _URL.format(model=model_name, data=DATA_TO_PATH[data]) + url = _URL.format(model=model_name, data=DATA_TO_PATH.get(data, data)) local_filename = url.split("/")[-1] location = os.path.join(_CACHE_DIR, local_filename) if os.path.exists(location) and not force: