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

Keyboard interrupt retains seed and log information in files produced prior to interrupt. Closes #21

Browse Source
This commit is contained in:
Lincoln Stein
2022-08-23 00:51:38 -04:00
parent bc7b1fdd37
commit 710b908290
2 changed files with 109 additions and 101 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
scripts/dream.py
View File

@ -85,7 +85,8 @@ def main():
cmd_parser = create_cmd_parser()
main_loop(t2i,cmd_parser,log,infile)
log.close()
infile.close()
if infile:
infile.close()
def main_loop(t2i,parser,log,infile):
@ -157,19 +158,12 @@ def main_loop(t2i,parser,log,infile):
print("Try again with a prompt!")
continue
try:
if opt.init_img is None:
results = t2i.txt2img(**vars(opt))
else:
results = t2i.img2img(**vars(opt))
print("Outputs:")
write_log_message(t2i,opt,results,log)
except KeyboardInterrupt:
print('*interrupted*')
continue
except RuntimeError as e:
print(str(e))
continue
if opt.init_img is None:
results = t2i.txt2img(**vars(opt))
else:
results = t2i.img2img(**vars(opt))
print("Outputs:")
write_log_message(t2i,opt,results,log)
print("goodbye!")