Mirrors
/
EveryDream
mirror of https://github.com/victorchall/EveryDream.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

autocreate cache and output folder

This commit is contained in:
Victor Hall 2022-11-04 13:48:05 -04:00
parent f343c054e1
commit b56373762c
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
scripts/auto_caption.py
View File

@ -89,8 +89,15 @@ async def main(opt):
config_path = "scripts/BLIP/configs/med_config.json" config_path = "scripts/BLIP/configs/med_config.json"
cache_folder = ".cache"
model_cache_path = ".cache/model_base_caption_capfilt_large.pth" model_cache_path = ".cache/model_base_caption_capfilt_large.pth"
if not os.path.exists(cache_folder):
os.makedirs(cache_folder)
if not os.path.exists(opt.out_dir):
os.makedirs(opt.out_dir)
if not os.path.exists(model_cache_path): if not os.path.exists(model_cache_path):
print(f"Downloading model to {model_cache_path}... please wait") print(f"Downloading model to {model_cache_path}... please wait")