From 630256d01b1a5fe06625fc85aed73b4c79dae901 Mon Sep 17 00:00:00 2001 From: Qing Date: Sat, 14 Jan 2023 22:08:45 +0800 Subject: [PATCH] fix model download issue --- lama_cleaner/helper.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lama_cleaner/helper.py b/lama_cleaner/helper.py index cac5378..16fa64e 100644 --- a/lama_cleaner/helper.py +++ b/lama_cleaner/helper.py @@ -15,7 +15,7 @@ def get_cache_path_by_url(url): hub_dir = get_dir() model_dir = os.path.join(hub_dir, "checkpoints") 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) cached_file = os.path.join(model_dir, filename) return cached_file