File size: 1,367 Bytes
a9806c1
1e8b518
 
93caa86
1e8b518
 
 
 
 
93caa86
9bba7bd
 
 
93caa86
9bba7bd
93caa86
9bba7bd
 
93caa86
1e8b518
 
 
 
 
93caa86
1e8b518
93caa86
1e8b518
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
import os
import shutil
import subprocess
print("\033[33m欢迎使用阿里云SD一键部署服务!预计时长5分钟,请耐心等待\033[0m")
subprocess.run(["python3", "-m", "pip", "install", "--upgrade", "pip"])
subprocess.run(["apt", "update"])
subprocess.run(["apt", "install", "-y", "aria2"])
subprocess.run(["apt-get", "install", "sudo"])
subprocess.run(["sudo", "apt", "install", "p7zip-full", "-y"])
print("\n\n\n\n\033[33m开始下载 stable-diffusion-webui.zip\033[0m")
def aria2(url, filename, d):
    subprocess.run(["aria2c", "--console-log-level=error", "-c", "-x", "16", "-s", "16", url, "-o", filename, "-d", d])
url = "https://www.modelscope.cn/api/v1/models/dong625/stable-diffusion-webui/repo?Revision=master&FilePath=stable-diffusion-webui.zip"
aria2(url, "stable-diffusion-webui.zip", "/mnt")
subprocess.run([
    "7z", "x", "-pQQ917724495", "/mnt/stable-diffusion-webui.zip",
    "-o/mnt/workspace", "-y"
])
os.remove("/mnt/stable-diffusion-webui.zip")
os.chdir("/mnt/workspace/stable-diffusion-webui")
subprocess.run([
    "pip", "install", "-r", "requirements_versions.txt",
    "--extra-index-url", "https://download.pytorch.org/whl/cu117"
])
subprocess.run(["python", "move.py"])
if __name__ == "__main__":
    print("\n\n\n\n\033[32m\033[4mSD服务部署成功!请点击左侧界面启动SD服务!\033[0m")
    os.remove(__file__)