From 25f8d87be8b2141477ddf18ae28fcbf6082fd74e Mon Sep 17 00:00:00 2001 From: Maja Jablonska <majajjablonska@gmail.com> Date: Wed, 15 Nov 2023 22:33:53 +1100 Subject: [PATCH] A small correction to url --- combo/utils/download.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/combo/utils/download.py b/combo/utils/download.py index 7a928d2..e88c759 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) -- GitLab