add --cuda-malloc, --disable-cuda-malloc cmd args

This commit is contained in:
Won-Kyu Park 2024-10-12 22:49:14 +09:00
parent 6668e8749b
commit e78be27e75
No known key found for this signature in database
GPG Key ID: 53AA79C8C9535D15

View File

@ -9,6 +9,9 @@ parser.add_argument("-f", action='store_true', help=argparse.SUPPRESS) # allows
parser.add_argument("--update-all-extensions", action='store_true', help="launch.py argument: download updates for all extensions when starting the program")
parser.add_argument("--skip-python-version-check", action='store_true', help="launch.py argument: do not check python version")
parser.add_argument("--skip-torch-cuda-test", action='store_true', help="launch.py argument: do not check if CUDA is able to work properly")
cm_group = parser.add_mutually_exclusive_group()
cm_group.add_argument("--cuda-malloc", action='store_true', help="Enable cudaMallocAsync (enabled by default for torch 2.0 and up).")
cm_group.add_argument("--disable-cuda-malloc", action="store_true", help="Disable cudaMallocAsync.")
parser.add_argument("--reinstall-xformers", action='store_true', help="launch.py argument: install the appropriate version of xformers even if you have some version already installed")
parser.add_argument("--reinstall-torch", action='store_true', help="launch.py argument: install the appropriate version of torch even if you have some version already installed")
parser.add_argument("--update-check", action='store_true', help="launch.py argument: check for updates at startup")