diff options
author | Greendayle <81877451+Greendayle@users.noreply.github.com> | 2022-10-08 18:28:22 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-10-08 18:28:22 +0200 |
commit | 0ec80f0125a14c03ac860279f40c0c062dbde0cf (patch) | |
tree | f51f04feecca9006a6a40e622bae3d6ae997e5eb /modules/shared.py | |
parent | 01f8cb44474e454903c11718e6a4f33dbde34bb8 (diff) | |
parent | 3061cdb7b610d4ba7f1ea695d9d6364b591e5bc7 (diff) |
Merge branch 'master' into dev/deepdanbooru
Diffstat (limited to 'modules/shared.py')
-rw-r--r-- | modules/shared.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/modules/shared.py b/modules/shared.py index c87b726e..026dce47 100644 --- a/modules/shared.py +++ b/modules/shared.py @@ -44,6 +44,7 @@ parser.add_argument("--scunet-models-path", type=str, help="Path to directory wi parser.add_argument("--swinir-models-path", type=str, help="Path to directory with SwinIR model file(s).", default=os.path.join(models_path, 'SwinIR'))
parser.add_argument("--ldsr-models-path", type=str, help="Path to directory with LDSR model file(s).", default=os.path.join(models_path, 'LDSR'))
parser.add_argument("--xformers", action='store_true', help="enable xformers for cross attention layers")
+parser.add_argument("--force-enable-xformers", action='store_true', help="enable xformers for cross attention layers regardless of whether the checking code thinks you can run it; do not make bug reports if this fails to work")
parser.add_argument("--deepdanbooru", action='store_true', help="enable deepdanbooru interrogator")
parser.add_argument("--opt-split-attention", action='store_true', help="force-enables cross-attention layer optimization. By default, it's on for torch.cuda and off for other torch devices.")
parser.add_argument("--disable-opt-split-attention", action='store_true', help="force-disables cross-attention layer optimization")
|