From 0ceabef4ecb29400dac31aae7a77ab19c36041ec Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C5=81ukasz=20Pszenny?= <pszenny@e-science.pl>
Date: Tue, 28 Nov 2023 19:33:48 +0000
Subject: [PATCH] Update download.py to support scene models
---
combo/utils/download.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/combo/utils/download.py b/combo/utils/download.py
index f5deb82..448a994 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:
--
GitLab