From 0947434ec37df4f7f691536defe6ac00be91d540 Mon Sep 17 00:00:00 2001 From: MSFTserver Date: Sun, 3 Apr 2022 22:45:12 -0700 Subject: [PATCH] update clip imports and path exists check --- disco.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/disco.py b/disco.py index c7d5b42..2137b02 100644 --- a/disco.py +++ b/disco.py @@ -444,11 +444,11 @@ if is_colab: subprocess.run(['apt', 'install', 'imagemagick'], stdout=subprocess.PIPE).stdout.decode('utf-8') try: - import clip + import CLIP except: if os.path.exists("CLIP") is not True: gitclone("https://github.com/openai/CLIP") - sys.path.append(f'{root_path}/CLIP') + sys.path.append(f'{PROJECT_DIR}/CLIP') try: from guided_diffusion.script_util import create_model_and_diffusion @@ -514,7 +514,7 @@ from torch.nn import functional as F import torchvision.transforms as T import torchvision.transforms.functional as TF from tqdm.notebook import tqdm -import clip +import CLIP from resize_right import resize from guided_diffusion.script_util import create_model_and_diffusion, model_and_diffusion_defaults from datetime import datetime @@ -545,7 +545,7 @@ if USE_ADABINS: except: if os.path.exists("AdaBins") is not True: gitclone("https://github.com/shariqfarooq123/AdaBins.git") - if not path_exists(f'{model_path}/pretrained/AdaBins_nyu.pt'): + 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')