From 753dbeae31939c0f748319feab1719ab69f91b31 Mon Sep 17 00:00:00 2001 From: Adam Letts Date: Tue, 5 Apr 2022 12:33:11 -0400 Subject: [PATCH] Fix AdaBins model installation --- Disco_Diffusion.ipynb | 8 ++++---- disco.py | 8 ++++---- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/Disco_Diffusion.ipynb b/Disco_Diffusion.ipynb index 5255544..6c1c809 100644 --- a/Disco_Diffusion.ipynb +++ b/Disco_Diffusion.ipynb @@ -577,10 +577,10 @@ " except:\n", " if os.path.exists(\"AdaBins\") is not True:\n", " gitclone(\"https://github.com/shariqfarooq123/AdaBins.git\")\n", - " if not os.path.exists(f'{model_path}/pretrained/AdaBins_nyu.pt'):\n", - " os.makedirs(f'{model_path}/pretrained')\n", - " wget(\"https://cloudflare-ipfs.com/ipfs/Qmd2mMnDLWePKmgfS8m6ntAg4nhV5VkUyAydYBp8cWWeB7/AdaBins_nyu.pt\", f'{model_path}/pretrained')\n", - " sys.path.append(f'{os.getcwd()}/AdaBins')\n", + " if not os.path.exists(f'{PROJECT_DIR}/pretrained/AdaBins_nyu.pt'):\n", + " createPath(f'{PROJECT_DIR}/pretrained')\n", + " wget("https://cloudflare-ipfs.com/ipfs/Qmd2mMnDLWePKmgfS8m6ntAg4nhV5VkUyAydYBp8cWWeB7/AdaBins_nyu.pt", f'{PROJECT_DIR}/pretrained')\n", + " sys.path.append(f'{PROJECT_DIR}/AdaBins')\n", " from infer import InferenceHelper\n", " MAX_ADABINS_AREA = 500000\n", "\n", diff --git a/disco.py b/disco.py index 92d712c..d34383c 100644 --- a/disco.py +++ b/disco.py @@ -549,10 +549,10 @@ if USE_ADABINS: except: if os.path.exists("AdaBins") is not True: gitclone("https://github.com/shariqfarooq123/AdaBins.git") - if not os.path.exists(f'{model_path}/pretrained/AdaBins_nyu.pt'): - os.makedirs(f'{model_path}/pretrained') - wget("https://cloudflare-ipfs.com/ipfs/Qmd2mMnDLWePKmgfS8m6ntAg4nhV5VkUyAydYBp8cWWeB7/AdaBins_nyu.pt", f'{model_path}/pretrained') - sys.path.append(f'{os.getcwd()}/AdaBins') + if not os.path.exists(f'{PROJECT_DIR}/pretrained/AdaBins_nyu.pt'): + createPath(f'{PROJECT_DIR}/pretrained') + wget("https://cloudflare-ipfs.com/ipfs/Qmd2mMnDLWePKmgfS8m6ntAg4nhV5VkUyAydYBp8cWWeB7/AdaBins_nyu.pt", f'{PROJECT_DIR}/pretrained') + sys.path.append(f'{PROJECT_DIR}/AdaBins') from infer import InferenceHelper MAX_ADABINS_AREA = 500000