Commit 746367b7 authored by 孙傲's avatar 孙傲

适配autoDL安装

parent ccd1a19e
...@@ -296,10 +296,10 @@ def prepare_environment(): ...@@ -296,10 +296,10 @@ def prepare_environment():
os.makedirs(os.path.join(script_path, dir_repos), exist_ok=True) os.makedirs(os.path.join(script_path, dir_repos), exist_ok=True)
git_clone(stable_diffusion_repo, repo_dir('stable-diffusion-stability-ai'), "Stable Diffusion", stable_diffusion_commit_hash) # git_clone(stable_diffusion_repo, repo_dir('stable-diffusion-stability-ai'), "Stable Diffusion", stable_diffusion_commit_hash)
git_clone(k_diffusion_repo, repo_dir('k-diffusion'), "K-diffusion", k_diffusion_commit_hash) # git_clone(k_diffusion_repo, repo_dir('k-diffusion'), "K-diffusion", k_diffusion_commit_hash)
git_clone(codeformer_repo, repo_dir('CodeFormer'), "CodeFormer", codeformer_commit_hash) # git_clone(codeformer_repo, repo_dir('CodeFormer'), "CodeFormer", codeformer_commit_hash)
git_clone(blip_repo, repo_dir('BLIP'), "BLIP", blip_commit_hash) # git_clone(blip_repo, repo_dir('BLIP'), "BLIP", blip_commit_hash)
if not is_installed("lpips"): if not is_installed("lpips"):
run_pip(f"install -r \"{os.path.join(repo_dir('CodeFormer'), 'requirements.txt')}\"", "requirements for CodeFormer") run_pip(f"install -r \"{os.path.join(repo_dir('CodeFormer'), 'requirements.txt')}\"", "requirements for CodeFormer")
......
...@@ -4,13 +4,13 @@ ...@@ -4,13 +4,13 @@
######################################################### #########################################################
# Install directory without trailing slash # Install directory without trailing slash
install_dir="/dev/shm/stable-diffusion/" install_dir="/root/stable-diffusion/"
# Name of the subdirectory # Name of the subdirectory
#clone_dir="stable-diffusion-webui" #clone_dir="stable-diffusion-webui"
# Commandline arguments for webui.py, for example: export COMMANDLINE_ARGS="--medvram --opt-split-attention" # Commandline arguments for webui.py, for example: export COMMANDLINE_ARGS="--medvram --opt-split-attention"
export COMMANDLINE_ARGS="--xformers" export COMMANDLINE_ARGS="--xformers --port 6006"
# python3 executable # python3 executable
#python_cmd="python3" #python_cmd="python3"
...@@ -32,7 +32,10 @@ export COMMANDLINE_ARGS="--xformers" ...@@ -32,7 +32,10 @@ export COMMANDLINE_ARGS="--xformers"
# Fixed git repos # Fixed git repos
#export K_DIFFUSION_PACKAGE="" #export K_DIFFUSION_PACKAGE=""
#export GFPGAN_PACKAGE="" export GFPGAN_PACKAGE="https://public.yitong.com/mirrors/github.com/TencentARC/GFPGAN/archive/8d2447a2d918f8eba5a4a01463fd48e45126a379.zip"
export CLIP_PACKAGE="https://public.yitong.com/mirrors/github.com/openai/CLIP/archive/d50d76daa670286dd6cacf3bcd80b5e4823fc8e1.zip"
export OPENCLIP_PACKAGE="https://public.yitong.com/mirrors/github.com/mlfoundations/open_clip/archive/bb6e834e9c70d9c27d0dc3ecedeebeaeb1ffad6b.zip"
# Fixed git commits # Fixed git commits
#export STABLE_DIFFUSION_COMMIT_HASH="" #export STABLE_DIFFUSION_COMMIT_HASH=""
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment