Skip to content

Commit 0ab5a36

Browse files
committed
fix missing lines in outputs
1 parent 5e43372 commit 0ab5a36

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

scripts/dream.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -321,6 +321,7 @@ def image_writer(image, seed, upscaled=False):
321321
print('Outputs:')
322322
log_path = os.path.join(current_outdir, 'dream_log.txt')
323323
write_log_message(results, log_path)
324+
print()
324325

325326
print('goodbye!')
326327

@@ -371,7 +372,8 @@ def write_log_message(results, log_path):
371372
log_lines = [f'{path}: {prompt}\n' for path, prompt in results]
372373
for l in log_lines:
373374
output_cntr += 1
374-
print(output_cntr)
375+
print(f'[{output_cntr}] {l}',end='')
376+
375377

376378
with open(log_path, 'a', encoding='utf-8') as file:
377379
file.writelines(log_lines)

0 commit comments

Comments
 (0)