michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

check for OS and env variable

Browse source
This commit is contained in:
C43H66N12O12S2 2022-10-08 16:13:26 +03:00 committed by GitHub
parent 3f166be1b6
commit d0e85873ac
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
launch.py
View file

@ -4,6 +4,7 @@ import os
import sys import sys
import importlib.util import importlib.util
import shlex import shlex
import platform
dir_repos = "repositories" dir_repos = "repositories"
dir_tmp = "tmp" dir_tmp = "tmp"
@ -31,6 +32,7 @@ def extract_arg(args, name):
args, skip_torch_cuda_test = extract_arg(args, '--skip-torch-cuda-test') args, skip_torch_cuda_test = extract_arg(args, '--skip-torch-cuda-test')
args, xformers = extract_arg(args, '--xformers')
def repo_dir(name): def repo_dir(name):
@ -124,8 +126,11 @@ if not is_installed("gfpgan"):
if not is_installed("clip"): if not is_installed("clip"):
run_pip(f"install {clip_package}", "clip") run_pip(f"install {clip_package}", "clip")
if not is_installed("xformers"): if not is_installed("xformers") and xformers:
run_pip("install https://github.com/C43H66N12O12S2/stable-diffusion-webui/releases/download/a/xformers-0.0.14.dev0-cp310-cp310-win_amd64.whl", "xformers") if platform.system() == "Windows":
run_pip("install https://github.com/C43H66N12O12S2/stable-diffusion-webui/releases/download/a/xformers-0.0.14.dev0-cp310-cp310-win_amd64.whl", "xformers")
elif:
run_pip("install xformers", "xformers")
os.makedirs(dir_repos, exist_ok=True) os.makedirs(dir_repos, exist_ok=True)