aboutsummaryrefslogtreecommitdiffstats
path: root/launch.py
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2022-10-02 15:21:56 +0000
committerAUTOMATIC <16777216c@gmail.com>2022-10-02 15:21:56 +0000
commit53a3dc601fb734ce433505b1ca68770919106bad (patch)
treeba1e3f67ef85a03acbc19177ca6b7ede18f6fe27 /launch.py
parent6e063124acc06c13e8bfae37b1f1cd372f3e04ab (diff)
downloadstable-diffusion-webui-gfx803-53a3dc601fb734ce433505b1ca68770919106bad.tar.gz
stable-diffusion-webui-gfx803-53a3dc601fb734ce433505b1ca68770919106bad.tar.bz2
stable-diffusion-webui-gfx803-53a3dc601fb734ce433505b1ca68770919106bad.zip
move CLIP out of requirements and into launcher to make it possible to launch the program offline
Diffstat (limited to 'launch.py')
-rw-r--r--launch.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/launch.py b/launch.py
index d2793ed2..57405fea 100644
--- a/launch.py
+++ b/launch.py
@@ -15,6 +15,7 @@ requirements_file = os.environ.get('REQS_FILE', "requirements_versions.txt")
commandline_args = os.environ.get('COMMANDLINE_ARGS', "")
gfpgan_package = os.environ.get('GFPGAN_PACKAGE', "git+https://github.com/TencentARC/GFPGAN.git@8d2447a2d918f8eba5a4a01463fd48e45126a379")
+clip_package = os.environ.get('CLIP_PACKAGE', "git+https://github.com/openai/CLIP.git@d50d76daa670286dd6cacf3bcd80b5e4823fc8e1")
stable_diffusion_commit_hash = os.environ.get('STABLE_DIFFUSION_COMMIT_HASH', "69ae4b35e0a0f6ee1af8bb9a5d0016ccb27e36dc")
taming_transformers_commit_hash = os.environ.get('TAMING_TRANSFORMERS_COMMIT_HASH', "24268930bf1dce879235a7fddd0b2355b84d7ea6")
@@ -111,6 +112,9 @@ if not skip_torch_cuda_test:
if not is_installed("gfpgan"):
run_pip(f"install {gfpgan_package}", "gfpgan")
+if not is_installed("clip"):
+ run_pip(f"install {clip_package}", "clip")
+
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)