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