|
|
import os |
|
|
import shutil |
|
|
import subprocess |
|
|
|
|
|
print("\033[36m\033[4mFooocus部署服务启动中，下载模型共65G，预计部署时长为50分钟\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"]) |
|
|
os.system("sudo apt-get install git-lfs") |
|
|
os.system("git lfs install") |
|
|
|
|
|
def aria2_download(url, filename, d): |
|
|
os.system(f"aria2c --console-log-level=error -c -x 16 -s 16 {url} -o {filename} -d {d}") |
|
|
|
|
|
def main(): |
|
|
url = "https://hf-mirror.com/dong625/Fooocus/resolve/main/Fooocus.zip" |
|
|
|
|
|
aria2_download(url, "Fooocus.zip", "/mnt/workspace/") |
|
|
os.system("7z x -pQQ917724495 /mnt/workspace/Fooocus.zip -o/mnt/workspace -y") |
|
|
os.remove("/mnt/workspace/Fooocus.zip") |
|
|
os.chdir("/mnt/workspace/Fooocus") |
|
|
os.system("pip install -r requirements_versions.txt") |
|
|
print("Environment success") |
|
|
subprocess.run(["python", "Predownload.py"]) |
|
|
subprocess.run(["python", "move.py"]) |
|
|
|
|
|
if __name__ == "__main__": |
|
|
main() |
|
|
print("\033[32m\033[4mFooocus服务部署成功！请点击左侧界面启动Fooocus服务！\033[0m") |
|
|
os.remove(__file__) |
