@ -42,7 +42,7 @@ parser.add_argument("--auto-launch", action="store_true", help="Automatically la
parser . add_argument ( " --cuda-device " , type = int , default = None , metavar = " DEVICE_ID " , help = " Set the id of the cuda device this instance will use. " )
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 = " En able cudaMallocAsync." )
cm_group . add_argument ( " --disable-cuda-malloc " , action = " store_true " , help = " Dis able cudaMallocAsync." )
parser . add_argument ( " --dont-upcast-attention " , action = " store_true " , help = " Disable upcasting of attention. Can boost speed but increase the chances of black images. " )