From 42ffcd7204d761a3752f79b53f97a07181ee071c Mon Sep 17 00:00:00 2001
From: Lincoln Stein <lincoln.stein@gmail.com>
Date: Mon, 29 Aug 2022 22:34:09 -0400
Subject: [PATCH] add the recently added commands to the readline
 command-line-completion list; fix command-line documentation bug, closing
 issue #188

---
 ldm/dream/readline.py | 47 +++++++++++++++++++------------------------
 scripts/dream.py      |  2 +-
 2 files changed, 22 insertions(+), 27 deletions(-)

diff --git a/ldm/dream/readline.py b/ldm/dream/readline.py
index f40fe83316..5cf99523fc 100644
--- a/ldm/dream/readline.py
+++ b/ldm/dream/readline.py
@@ -86,32 +86,27 @@ if readline_available:
     readline.set_completer(
         Completer(
             [
-                'cd',
-                'pwd',
-                '--steps',
-                '-s',
-                '--seed',
-                '-S',
-                '--iterations',
-                '-n',
-                '--batch_size',
-                '-b',
-                '--width',
-                '-W',
-                '--height',
-                '-H',
-                '--cfg_scale',
-                '-C',
-                '--grid',
-                '-g',
-                '--individual',
-                '-i',
-                '--init_img',
-                '-I',
-                '--strength',
-                '-f',
-                '-v',
-                '--variants',
+                '--steps','-s',
+                '--seed','-S',
+                '--iterations','-n',
+                '--batch_size','-b',
+                '--width','-W','--height','-H',
+                '--cfg_scale','-C',
+                '--grid','-g',
+                '--individual','-i',
+                '--init_img','-I',
+                '--strength','-f',
+                '--variants','-v',
+                '--outdir','-o',
+                '--sampler','-A','-m',
+                '--embedding_path',
+                '--device',
+                '--grid','-g',
+                '--gfpgan_strength','-G',
+                '--upscale','-U',
+                '-save_orig','--save_original',
+                '--skip_normalize','-x',
+                '--log_tokenization','t',
             ]
         ).complete
     )
diff --git a/scripts/dream.py b/scripts/dream.py
index ae79b76c17..0f4dbdae06 100755
--- a/scripts/dream.py
+++ b/scripts/dream.py
@@ -437,7 +437,7 @@ def create_cmd_parser():
         '--cfg_scale',
         default=7.5,
         type=float,
-        help='Prompt configuration scale',
+        help='Classifier free guidance (CFG) scale - higher numbers cause generator to "try" harder.',
     )
     parser.add_argument(
         '-g', '--grid', action='store_true', help='generate a grid'