diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2022-10-09 07:52:21 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-10-09 07:52:21 +0000 |
| commit | e00b4df7c6f0a13941d6f6ea425eebdaa2bc9318 (patch) | |
| tree | c01de5d0efb2f6bb414722e398e29316ecafb74f /launch.py | |
| parent | 14192c5b207b16b1ec7a4c9c4ea538d1a6811a4d (diff) | |
| parent | 0ec80f0125a14c03ac860279f40c0c062dbde0cf (diff) | |
| download | stable-diffusion-webui-gfx803-e00b4df7c6f0a13941d6f6ea425eebdaa2bc9318.tar.gz stable-diffusion-webui-gfx803-e00b4df7c6f0a13941d6f6ea425eebdaa2bc9318.tar.bz2 stable-diffusion-webui-gfx803-e00b4df7c6f0a13941d6f6ea425eebdaa2bc9318.zip | |
Merge pull request #1752 from Greendayle/dev/deepdanbooru
Added DeepDanbooru interrogator
Diffstat (limited to 'launch.py')
| -rw-r--r-- | launch.py | 4 |
1 files changed, 4 insertions, 0 deletions
@@ -33,6 +33,7 @@ def extract_arg(args, name): args, skip_torch_cuda_test = extract_arg(args, '--skip-torch-cuda-test')
xformers = '--xformers' in args
+deepdanbooru = '--deepdanbooru' in args
def repo_dir(name):
@@ -132,6 +133,9 @@ if not is_installed("xformers") and xformers and platform.python_version().start elif platform.system() == "Linux":
run_pip("install xformers", "xformers")
+if not is_installed("deepdanbooru") and deepdanbooru:
+ run_pip("install git+https://github.com/KichangKim/DeepDanbooru.git@edf73df4cdaeea2cf00e9ac08bd8a9026b7a7b26#egg=deepdanbooru[tensorflow] tensorflow==2.10.0 tensorflow-io==0.27.0", "deepdanbooru")
+
os.makedirs(dir_repos, exist_ok=True)
git_clone("https://github.com/CompVis/stable-diffusion.git", repo_dir('stable-diffusion'), "Stable Diffusion", stable_diffusion_commit_hash)
|