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

Bug Fixes

Browse Source
This commit is contained in:
blessedcoolant
2022-09-17 14:12:35 +12:00
parent 61f46cac31
commit 42072fc15c
2 changed files with 17 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
scripts/dream.py
View File

@ -123,7 +123,7 @@ def main_loop(gen, opt, infile):
if command.startswith(('#', '//')):
continue
if command.startswith('q '):
if len(command.strip()) == 1 and command.startswith('q'):
done = True
break
@ -138,7 +138,7 @@ def main_loop(gen, opt, infile):
parser.print_help()
continue
if len(opt.prompt) == 0:
print('Try again with a prompt!')
print('\nTry again with a prompt!')
continue
# retrieve previous value!
@ -191,14 +191,14 @@ def main_loop(gen, opt, infile):
if not os.path.exists(opt.outdir):
os.makedirs(opt.outdir)
current_outdir = opt.outdir
elif prompt_as_dir:
elif opt.prompt_as_dir:
# sanitize the prompt to a valid folder name
subdir = path_filter.sub('_', opt.prompt)[:name_max].rstrip(' .')
# truncate path to maximum allowed length
# 27 is the length of '######.##########.##.png', plus two separators and a NUL
subdir = subdir[:(path_max - 27 - len(os.path.abspath(opt.outdir)))]
current_outdir = os.path.join(outdir, subdir)
current_outdir = os.path.join(opt.outdir, subdir)
print('Writing files to directory: "' + current_outdir + '"')
@ -206,7 +206,7 @@ def main_loop(gen, opt, infile):
if not os.path.exists(current_outdir):
os.makedirs(current_outdir)
else:
current_outdir = outdir
current_outdir = opt.outdir
# Here is where the images are actually generated!
last_results = []