add --karras_max option to invoke.py command line (#1762)

This addresses image regression image reported in #1754
This commit is contained in:
Lincoln Stein 2022-12-12 13:16:15 -05:00 committed by GitHub
parent 5c3cbd05f1
commit 7314f1a862
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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',