Skip to content
Snippets Groups Projects
Commit 0ceabef4 authored by Łukasz Pszenny's avatar Łukasz Pszenny
Browse files

Update download.py to support scene models

parent acb98ee6
No related branches found
No related tags found
1 merge request!46Merge COMBO 3.0 into master
......@@ -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:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment