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

refactored logfile handling; minimised time spent in context managers (with open)

Browse Source
This commit is contained in:
yun saki 2022-08-26 13:22:53 +02:00
parent 12f59e1daa
commit e00397f9ca
1 changed files with 10 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
scripts/dream.py
View File

@ -118,12 +118,11 @@ def main():
) )
log_path = os.path.join(opt.outdir, 'dream_log.txt') log_path = os.path.join(opt.outdir, 'dream_log.txt')
with open(log_path, 'a') as log: cmd_parser = create_cmd_parser()
cmd_parser = create_cmd_parser() main_loop(t2i, opt.outdir, cmd_parser, log_path, infile)
main_loop(t2i, opt.outdir, cmd_parser, log, infile)
def main_loop(t2i, outdir, parser, log, infile): def main_loop(t2i, outdir, parser, log_path, infile):
"""prompt/read/execute loop""" """prompt/read/execute loop"""
done = False done = False
last_seeds = [] last_seeds = []
@ -240,7 +239,7 @@ def main_loop(t2i, outdir, parser, log, infile):
continue continue
print('Outputs:') print('Outputs:')
write_log_message(t2i, normalized_prompt, results, log) write_log_message(t2i, normalized_prompt, results, log_path)
print('goodbye!') print('goodbye!')
@ -310,19 +309,14 @@ def load_gfpgan_bg_upsampler(bg_upsampler, bg_tile=400):
# return variants # return variants
def write_log_message(t2i, prompt, results, logfile): ### the results variable doesn't seem to be necessary. maybe remove it?
def write_log_message(t2i, prompt, results, log_path):
"""logs the name of the output image, its prompt and seed to the terminal, log file, and a Dream text chunk in the PNG metadata""" """logs the name of the output image, its prompt and seed to the terminal, log file, and a Dream text chunk in the PNG metadata"""
last_seed = None log_lines = [f"{r[0]}: {prompt} -S{seed}\n" for r in results]
img_num = 1 print(*log_lines, sep="")
seenit = {}
for r in results: with open(log_path, "a") as file:
seed = r[1] file.writelines(log_lines)
log_message = f'{r[0]}: {prompt} -S{seed}'
print(log_message)
logfile.write(log_message + '\n')
logfile.flush()
def create_argv_parser(): def create_argv_parser():