Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Test for the presence of the specified img2img

Browse Source
This commit is contained in:
Sean McLellan 2022-08-23 19:22:35 -04:00
parent 0cdf5e61b0
commit 1df1e5c38b
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
scripts/dream.py
View File

@ -161,7 +161,12 @@ def main_loop(t2i,parser,log,infile):
if opt.init_img is None:
results = t2i.txt2img(**vars(opt))
else:
results = t2i.img2img(**vars(opt))
if os.path.exists(opt.init_img):
results = t2i.img2img(**vars(opt))
else:
print(f"An image was not present at {opt.init_img}. Try again with a valid image path!")
continue
print("Outputs:")
write_log_message(t2i,opt,results,log)