import sys #line:1 import os #line:2 from IPython import get_ipython #line:3 from IPython .utils import capture #line:4 from IPython .display import clear_output #line:5 wb =f'/home/studio-lab-user/stable-diffusion-webui'#line:8 def gitDownload (O0O0OO0O00O00O0OO ,O0O00OO00O0O000OO ):#line:11 if os .path .exists (O0O00OO00O0O000OO ):#line:12 return #line:13 get_ipython ().system (f'git clone {O0O0OO0O00O00O0OO} {O0O00OO00O0O000OO}')#line:15 def installAdditional ():#line:18 OOO0O0O0000O0000O =['https://github.com/AlUlkesh/stable-diffusion-webui-images-browser','https://github.com/etherealxx/batchlinks-webui','https://github.com/camenduru/sd-civitai-browser','https://github.com/thomasasfk/sd-webui-aspect-ratio-helper','https://github.com/fkunn1326/openpose-editor','https://github.com/nonnonstop/sd-webui-3d-open-pose-editor','https://github.com/Physton/sd-webui-prompt-all-in-one','https://github.com/Bing-su/adetailer','https://github.com/picobyte/stable-diffusion-webui-wd14-tagger','https://github.com/Mikubill/sd-webui-controlnet','https://github.com/IDEA-Research/DWPose','https://github.com/Tencent/LightDiffusionFlow','https://github.com/continue-revolution/sd-webui-segment-anything','https://github.com/hanamizuki-ai/stable-diffusion-webui-localization-zh_Hans','https://github.com/pkuliyi2015/multidiffusion-upscaler-for-automatic1111',]#line:59 for OOOO00O0000O0OOOO in OOO0O0O0000O0000O :#line:60 OO0O0OOOO0O0O0O0O =OOOO00O0000O0OOOO .split ('/')[-1 ]#line:62 if 'github'in OOOO00O0000O0OOOO :#line:64 get_ipython ().system (f'git clone --depth 1 {OOOO00O0000O0OOOO} {wb}/extensions/{OO0O0OOOO0O0O0O0O}')#line:65 get_ipython ().system (f'wget https://raw.githubuserhome/studio-lab-user.com/camenduru/stable-diffusion-webui-scripts/main/run_n_times.py -O {wb}/scripts/run_n_times.py')#line:67 clear_output ()#line:69 gitDownload (f'https://huggingface.co/embed/negative',f'{wb}/embeddings/negative')#line:71 get_ipython ().system (f'rm -rf {wb}/embeddings/negative/.git')#line:72 get_ipython ().system (f'rm {wb}/embeddings/negative/.gitattributes')#line:73 gitDownload (f'https://huggingface.co/embed/lora',f'{wb}/models/Lora/positive')#line:75 get_ipython ().system (f'rm -rf {wb}/models/Lora/positive/.git')#line:76 get_ipython ().system (f'rm {wb}/models/Lora/positive/.gitattributes')#line:77 gitDownload (f'https://github.com/DominikDoom/a1111-sd-webui-tagcomplete',f'{wb}/extensions/a1111-sd-webui-tagcomplete')#line:80 get_ipython ().system (f'rm -rf {wb}/extensions/a1111-sd-webui-tagcomplete/tags/danbooru.csv')#line:81 get_ipython ().system (f'wget https://huggingface.co/gmk123/colab/raw/main/danbooru.csv -O {wb}/extensions/a1111-sd-webui-tagcomplete/tags/danbooru.csv')#line:82 gitDownload (f'https://github.com/s0md3v/sd-webui-roop',f'{wb}/extensions/sd-webui-roop')#line:85 get_ipython ().system (f'pip install -q insightface==0.7.3 onnx')#line:86 get_ipython ().system (f'mkdir -p {wb}/models/roop')#line:87 get_ipython ().system (f'wget https://huggingface.co/ezioruan/inswapper_128.onnx/resolve/main/inswapper_128.onnx -O {wb}/models/roop/inswapper_128.onnx')#line:88 print ('Roop换脸启用')#line:89 import subprocess #line:91 def sync_tmp_folder ():#line:93 O0O00O0O0OOO00O00 =[['rm','-rf','~/tmp/*','~/tmp','~/stable-diffusion-webui/models/Stable-diffusion/tmp_models','~/stable-diffusion-webui/models/ControlNet','~/stable-diffusion-webui/models/sam'],['ln','-vs','/tmp','~/tmp'],['ln','-vs','/tmp/models','~/stable-diffusion-webui/models/Stable-diffusion/tmp_models'],['ln','-vs','/tmp/ControlNet','~/stable-diffusion-webui/models/ControlNet'],['ln','-vs','/tmp/sam','~/stable-diffusion-webui/models/sam']]#line:100 for O0000O00O00OOO0OO in O0O00O0O0OOO00O00 :#line:102 O0OOOOOO00OOO0000 =[os .path .expanduser (OOOO00OO00OO0O0OO )for OOOO00OO00OO0O0OO in O0000O00O00OOO0OO ]#line:103 subprocess .run (O0OOOOOO00OOO0000 ,check =True )#line:104 def run_command (O00O0OOOOOOO0O0OO ,OO000OO0OOOO00O00 ):#line:107 print (f"Please wait, {OO000OO0OOOO00O00} is loading...")#line:108 subprocess .run (O00O0OOOOOOO0O0OO ,shell =True ,stdout =subprocess .DEVNULL ,stderr =subprocess .DEVNULL )#line:109 print (f"{OO000OO0OOOO00O00} ok!")#line:110 def zrok ():#line:113 try :#line:114 OOO0OO00000O000OO ="/home/studio-lab-user/.zrok/bin"#line:115 os .makedirs (OOO0OO00000O000OO ,exist_ok =True )#line:116 subprocess .run (["curl","-sLO","https://github.com/openziti/zrok/releases/download/v0.4.23/zrok_0.4.23_linux_amd64.tar.gz"],check =True )#line:117 subprocess .run (["tar","-xzf","zrok_0.4.23_linux_amd64.tar.gz","-C",OOO0OO00000O000OO ,"--wildcards","*zrok"],check =True )#line:118 os .remove ("zrok_0.4.23_linux_amd64.tar.gz")#line:119 except Exception as O000000OOO0000OOO :#line:121 print (f"An error occurred: {O000000OOO0000OOO}")#line:122 def initLocal ():#line:126 run_command ('conda install -y conda glib psutil gperftools aria2 gdown wget','【 Installing Conda 】')#line:128 run_command ('conda install -y -n base python=3.10.12','【 Installing Python 3.10 】')#line:129 run_command ('conda clean -y --all','【 Cleaning Conda 】')#line:130 run_command ('pip install torch==2.1.2+cu121 torchvision==0.16.2+cu121 torchaudio==2.1.2 --extra-index-url https://download.pytorch.org/whl/cu121','【 Installing Torch 】')#line:131 run_command ('pip install xformers==0.0.23.post1 triton==2.1.0','【 Installing xformers 】')#line:132 get_ipython ().system ('pip install -q tqdm')#line:134 clear_output ()#line:136 get_ipython ().run_line_magic ('cd','/home/studio-lab-user')#line:137 with capture .capture_output ()as OOOOOO000OOOOOOOO :#line:140 def OOOO00000000OO00O (O00000O00O0O0OO0O ,OO00O0O00O00O0OOO ,OOO00OO00OOOO00O0 ):OO0O0O0OO0OO00OO0 =widgets .Button (description =O00000O00O0O0OO0O ,disabled =True ,button_style =OO00O0O00O00O0OOO ,layout =widgets .Layout (min_width =OOO00OO00OOOO00O0 ));display (OO0O0O0OO0OO00OO0 )#line:141 get_ipython ().system (f'git clone -q --branch master https://github.com/AUTOMATIC1111/stable-diffusion-webui {wb}')#line:143 with capture .capture_output ()as OOOOOO000OOOOOOOO :#line:145 get_ipython ().run_line_magic ('cd',f'{wb}')#line:146 get_ipython ().system ('pip install -q -r requirements.txt')#line:148 get_ipython ().system (f'pip install -q basicsr')#line:150 get_ipython ().system (f'pip install -q fake_useragent')#line:151 sync_tmp_folder ()#line:154 clear_output ()#line:156 zrok ()#line:160 get_ipython ().system (f'wget https://huggingface.co/gmk123/run/raw/main/zrun.py -O {wb}/zrun.py')#line:161 get_ipython ().system (f'wget https://huggingface.co/gmk123/run/raw/main/zrok.py -O {wb}/zrok.py')#line:162 installAdditional ()#line:165 get_ipython ().system (f'rm -rf {wb}/config.json')#line:168 get_ipython ().system (f'wget -O {wb}/config.json "https://huggingface.co/gmk123/colab/raw/main/config.json"')#line:169 get_ipython ().system (f'wget -O {wb}/styles.csv "https://huggingface.co/gmk123/colab/raw/main/styles.csv"')#line:170 clear_output ()#line:172 get_ipython ().system (f'aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://civitai.com/api/download/models/192916 -d {wb}/models/Stable-diffusion -o MoyouArtificial_v1060.safetensors')#line:174 get_ipython ().system (f'aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.safetensors -d {wb}/models/VAE -o vae-ft-mse-840000-ema-pruned.safetensors')#line:176 get_ipython ().system (f'aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/embed/upscale/resolve/main/4x-UltraSharp.pth -d {wb}/models/ESRGAN -o 4x-UltraSharp.pth')#line:178 wise ="--xformers --enable-insecure-extension-access --theme dark --disable-console-progressbars --upcast-sampling"#line:180 ngrok_token =os .environ .get ('NGROK_TOKEN')#line:183 wise +=f" --ngrok {ngrok_token}"#line:184 def run (OO00000OO0O0O0O0O ):#line:187 clear_output ()#line:188 print (f'建议使用第二步：zrok部署，无需魔法，没有流量限制')#line:189 print (f'Ngrok注意事项：每个账号每月1G流量！需配合魔法使用')#line:190 get_ipython ().system (f'mkdir -p /tmp/ControlNet')#line:191 get_ipython ().run_line_magic ('cd',f'{wb}')#line:193 get_ipython ().system (f'python {OO00000OO0O0O0O0O} {wise}')#line:194 if os .path .exists (f'{wb}'):#line:197 run (f'webui.py')#line:199 else :#line:200 initLocal ()#line:202 run (f'launch.py')#line:205