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

fix missing lines in outputs

Browse Source
This commit is contained in:
Lincoln Stein 2022-09-11 17:19:46 -04:00
parent 5e433728b5
commit 0ab5a36464
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/dream.py
View File

@ -321,6 +321,7 @@ def main_loop(t2i, outdir, prompt_as_dir, parser, infile):
print('Outputs:') print('Outputs:')
log_path = os.path.join(current_outdir, 'dream_log.txt') log_path = os.path.join(current_outdir, 'dream_log.txt')
write_log_message(results, log_path) write_log_message(results, log_path)
print()
print('goodbye!') print('goodbye!')
@ -371,7 +372,8 @@ def write_log_message(results, log_path):
log_lines = [f'{path}: {prompt}\n' for path, prompt in results] log_lines = [f'{path}: {prompt}\n' for path, prompt in results]
for l in log_lines: for l in log_lines:
output_cntr += 1 output_cntr += 1
print(output_cntr) print(f'[{output_cntr}] {l}',end='')
with open(log_path, 'a', encoding='utf-8') as file: with open(log_path, 'a', encoding='utf-8') as file:
file.writelines(log_lines) file.writelines(log_lines)