From 7314f1a862ab5f2ec7340c3918259409efd67931 Mon Sep 17 00:00:00 2001
From: Lincoln Stein <lincoln.stein@gmail.com>
Date: Mon, 12 Dec 2022 13:16:15 -0500
Subject: [PATCH] add --karras_max option to invoke.py command line (#1762)

This addresses image regression image reported in #1754
---
 ldm/invoke/args.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/ldm/invoke/args.py b/ldm/invoke/args.py
index 4d5ea7b5cc..98a75c9886 100644
--- a/ldm/invoke/args.py
+++ b/ldm/invoke/args.py
@@ -583,6 +583,12 @@ class Args(object):
             action='store_true',
             help='Generates debugging image to display'
         )
+        render_group.add_argument(
+            '--karras_max',
+            type=int,
+            default=None,
+            help="control the point at which the K* samplers will shift from using the Karras noise schedule (good for low step counts) to the LatentDiffusion noise schedule (good for high step counts). Set to 0 to use LatentDiffusion for all step values, and to a high value (e.g. 1000) to use Karras for all step values. [29]."
+        )
         # Restoration related args
         postprocessing_group.add_argument(
             '--no_restore',