File size: 2,258 Bytes
f2d1e24 |
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 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 |
import subprocess
import concurrent.futures
from IPython.display import clear_output
import os
import sys
params = {}
for arg in sys.argv[1:]:
if arg.startswith('--'):
key_value = arg[len('--'):].split('=')
if len(key_value) == 2:
key, value = key_value
params[key] = value
subprocess.run(f'git clone https://github.com/Lucysck/test1 {params["dir"]}', shell=True)
def run_git():
subprocess.run(f'cp -rf /content/drive/MyDrive/sd/* {params["dir"]}', shell=True)
def run_aria2c():
subprocess.run(f'aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/spaces/weo1101/111/resolve/main/chilloutmix_NiPrunedFp32Fix-inpainting.inpainting.safetensors -d {params["dir"]}/models/Stable-diffusion -o chilloutmix_NiPrunedFp32Fix-inpainting.inpainting.safetensors', shell=True)
def run_code():
subprocess.run(f"wget -O {params['dir']}/libtcmalloc_minimal.so.4 https://huggingface.co/Vanwise/sd-colab/resolve/main/libtcmalloc_minimal.so.4", shell=True)
os.environ['LD_PRELOAD'] = f"{params['dir']}/libtcmalloc_minimal.so.4"
executor = concurrent.futures.ThreadPoolExecutor(max_workers=3)
task1 = executor.submit(run_git)
task2 = executor.submit(run_aria2c)
task3 = executor.submit(run_code)
concurrent.futures.wait([task1, task2, task3])
subprocess.run(f"wget -O {params['dir']}/config.json https://huggingface.co/spaces/weo1101/111/resolve/main/config.json", shell=True)
import threading
import time
def clear_output_pro():
clear_output()
print("Clearing output...")
threading.Timer(30, clear_output_pro).start()
threading.Timer(360, clear_output_pro).start()
os.chdir(f'{params["dir"]}')
clear_output()
base_arg = "--disable-safe-unpickle --opt-sdp-attention --no-half-vae --enable-insecure-extension --theme=dark --lowram --listen --xformers"
if params["share"] == "True":
base_arg += " --share"
if params["ngrok"]:
base_arg += f' --ngrok={params["ngrok"]} --ngrok-region="auto"'
if params["cloudflared"] == "True":
base_arg += f" --cloudflared"
if params["localhostrun"] == "True":
base_arg += f" --localhostrun"
if params["remotemoe"] == "True":
base_arg += f" --remotemoe"
subprocess.run(f"python launch.py {base_arg}", shell=True) |