diff --git a/ldm/dream/args.py b/ldm/dream/args.py index 6689fbb322..fd0f01390b 100644 --- a/ldm/dream/args.py +++ b/ldm/dream/args.py @@ -81,7 +81,7 @@ with metadata_from_png(): """ import argparse -from argparse import Namespace +from argparse import Namespace, RawTextHelpFormatter import shlex import json import hashlib @@ -463,7 +463,9 @@ class Args(object): # This creates the parser that processes commands on the dream> command line def _create_dream_cmd_parser(self): parser = argparse.ArgumentParser( - description=""" + formatter_class=RawTextHelpFormatter, + description= + """ *Image generation:* To generate images, type a text prompt with optional switches. Example: a fantastic alien landscape -W576 -H512 -s60 -n4 diff --git a/ldm/dream/log.py b/ldm/dream/log.py index beca10fa3f..8aebe62671 100644 --- a/ldm/dream/log.py +++ b/ldm/dream/log.py @@ -22,6 +22,8 @@ def write_log(results, log_path, file_types, output_cntr): def write_log_message(results, output_cntr): """logs to the terminal""" + if len(results) == 0: + return output_cntr log_lines = [f"{path}: {prompt}\n" for path, prompt in results] if len(log_lines)>1: subcntr = 1