| import os | |
| def ControlnetDownloads(): | |
| def aria2(url, filename, directory): | |
| os.system(f'aria2c --console-log-level=error -c -x 16 -s 16 {url} -o {filename} -d {directory}') | |
| print("\n\n\n\033[35m正在下载controlnet模型\033[0m") | |
| url = '"https://hf-mirror.com/dong625/ControlNet/resolve/main/ControlNet.zip"' | |
| aria2(url, "ControlNet.zip", "/mnt/workspace/stable-diffusion-webui/models") | |
| os.system('7z x -pQQ917724495 /mnt/workspace/stable-diffusion-webui/models/ControlNet.zip -o/mnt/workspace/stable-diffusion-webui/models -y') | |
| os.system('rm -rf /mnt/workspace/stable-diffusion-webui/models/ControlNet.zip') | |
| print("\n\n\n\033[35m正在下载controlnet预处理器\033[0m") | |
| url = '"https://hf-mirror.com/dong625/downloads/resolve/main/downloads.zip"' | |
| aria2(url, "downloads.zip", "/mnt/workspace/stable-diffusion-webui/extensions") | |
| os.system('7z x -pQQ917724495 /mnt/workspace/stable-diffusion-webui/extensions/downloads.zip -o/mnt/workspace/stable-diffusion-webui/extensions/sd-webui-controlnet/annotator -y') | |
| os.system('rm -rf /mnt/workspace/stable-diffusion-webui/extensions/downloads.zip') | |
| def main(): | |
| ControlnetDownloads() | |
| print("success") | |
| if __name__ == "__main__": | |
| main() | |