hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-05 03:32:37 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

some extra changes for collab compatibility

Browse Source
This commit is contained in:
AUTOMATIC
2022-09-06 10:38:02 +03:00
parent 395f170597
commit c688d0984e
2 changed files with 12 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
modules/shared.py
View File

@@ -32,7 +32,11 @@ parser.add_argument("--share", action='store_true', help="use share=True for gra
parser.add_argument("--esrgan-models-path", type=str, help="path to directory with ESRGAN models", default=os.path.join(script_path, 'ESRGAN'))
parser.add_argument("--opt-split-attention", action='store_true', help="enable optimization that reduced vram usage by a lot for about 10% decrease in performance")
parser.add_argument("--listen", action='store_true', help="launch gradio with 0.0.0.0 as server name, allowing to respond to network requests")
cmd_opts = parser.parse_args()
if 'notebook_argv' in globals():
cmd_opts = parser.parse_args(notebook_argv)
else:
cmd_opts = parser.parse_args()
cpu = torch.device("cpu")
gpu = torch.device("cuda")