aboutsummaryrefslogtreecommitdiff
path: root/modules/cmd_args.py
diff options
context:
space:
mode:
authorpapuSpartan <macabeg@icloud.com>2023-04-01 02:07:08 -0500
committerpapuSpartan <macabeg@icloud.com>2023-04-01 02:07:08 -0500
commit56680cd84ab68a283772cf697f8a72408a3f4167 (patch)
treed250aac38feb7b82b09c76f3266884f5f3ec421f /modules/cmd_args.py
parent22bcc7be428c94e9408f589966c2040187245d81 (diff)
first
Diffstat (limited to 'modules/cmd_args.py')
-rw-r--r--modules/cmd_args.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/modules/cmd_args.py b/modules/cmd_args.py
index 81c0b82a..4314f97b 100644
--- a/modules/cmd_args.py
+++ b/modules/cmd_args.py
@@ -101,3 +101,7 @@ parser.add_argument("--no-gradio-queue", action='store_true', help="Disables gra
parser.add_argument("--skip-version-check", action='store_true', help="Do not check versions of torch and xformers")
parser.add_argument("--no-hashing", action='store_true', help="disable sha256 hashing of checkpoints to help loading performance", default=False)
parser.add_argument("--no-download-sd-model", action='store_true', help="don't download SD1.5 model even if no model is found in --ckpt-dir", default=False)
+
+# token merging / tomesd
+parser.add_argument("--token-merging", action='store_true', help="Provides generation speedup by merging redundant tokens. (compatible with --xformers)", default=False)
+parser.add_argument("--token-merging-ratio", type=float, help="Adjusts ratio of merged to untouched tokens. Range: (0.0-1.0], Defaults to 0.5", default=0.5)