Update su.py

su.py

@@ -1,7 +1,6 @@
 import os
 import shutil
 import subprocess
-import modelscope
 
 subprocess.run(["python3", "-m", "pip", "install", "--upgrade", "pip"])
 subprocess.run(["apt", "update"])
@@ -10,9 +9,18 @@ subprocess.run(["apt-get", "install", "sudo"])
 subprocess.run(["sudo", "apt", "install", "p7zip-full", "-y"])
 os.system("sudo apt-get install git-lfs")
 os.system("git lfs install")
-print("stable-diffusion-webui.zip")
-os.system("git clone https://www.modelscope.cn/dong625/su.git")
-os.rename("su","stable-diffusion-webui")
+print("开始下载 stable-diffusion-webui.zip")
+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/workspace")
+#"https://hf-mirror.com/dong625/stable-diffusion-webui/resolve/main/stable-diffusion-webui.zip"
+subprocess.run([
+    "7z", "x", "-pQQ917724495", "/mnt/workspace/stable-diffusion-webui.zip",
+    "-o/mnt/workspace", "-y"
+])
+os.remove("/mnt/workspace/stable-diffusion-webui.zip")
+
 os.chdir("/mnt/workspace/stable-diffusion-webui")
 subprocess.run([
     "pip", "install", "-r", "requirements_versions.txt",
@@ -24,8 +32,8 @@ def move_files():
         "/mnt/workspace/stable-diffusion-webui/点我启动SD.ipynb",
         "/mnt/workspace/stable-diffusion-webui/下载大模型点这里.ipynb",
         "/mnt/workspace/stable-diffusion-webui/下载Lora点这里.ipynb",
-        "/mnt/workspace/stable-diffusion-webui/更多服务点我",
         "/mnt/workspace/stable-diffusion-webui/Photo",
+        "/mnt/workspace/stable-diffusion-webui/更多服务点我",
     ]
     destination_dir = "/mnt/workspace/"
     for file_path in source_files: