kamp/IOPaint
1
0
Fork 0
Code Issues Pull Requests Releases 1 Activity

fix model download issue

Browse Source
This commit is contained in:
Qing 2023-01-14 22:08:45 +08:00
parent 7d00fc8ceb
commit 630256d01b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lama_cleaner/helper.py
View File

@ -15,7 +15,7 @@ def get_cache_path_by_url(url):
hub_dir = get_dir() hub_dir = get_dir()
model_dir = os.path.join(hub_dir, "checkpoints") model_dir = os.path.join(hub_dir, "checkpoints")
if not os.path.isdir(model_dir): if not os.path.isdir(model_dir):
os.makedirs(os.path.join(model_dir, "hub", "checkpoints")) os.makedirs(model_dir)
filename = os.path.basename(parts.path) filename = os.path.basename(parts.path)
cached_file = os.path.join(model_dir, filename) cached_file = os.path.join(model_dir, filename)
return cached_file return cached_file