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
Branches
Tags
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% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment