stable-diffusion-easylized-kaggle-version.ipynb

{"cells":[{"cell_type":"code","execution_count":null,"metadata":{"id":"uCU2n8LGBqPx","trusted":true},"outputs":[],"source":["#@title ### 1. Configuration\n","\n","#@markdown # 1. models\n","official = False\n","anything = False\n","nai = False\n","waifu = False\n","counterfait = False\n","\n","#@markdown # 2. additional features\n","xformers = \"from_pypi\" # options: [\"from_source\", \"from_pypi\", \"none\"]\n","additional_server_parameters = \"--listen --xformers --enable-insecure-extension-access --theme dark --gradio-queue\"\n","disabled_extensions = []\n","\n","#@markdown # 3. debug options\n","#@markdown #### (if the code suddenly stops running and displays ^C while it is loading weights, tick the following box)\n","oHNo_My_SErVEr_keEpS_cRaShiNg_iTSeLF_HeLP_mE = True"]},{"cell_type":"code","execution_count":null,"metadata":{"id":"sBbcB4vwj_jm","trusted":true},"outputs":[],"source":["#@title ### 2. Installation\n","\n","from IPython.display import clear_output\n","%cd /kaggle/working/\n","!mkdir -p /kaggle/working/webui/ /kaggle/working/buildspace/ /kaggle/working/tools/ /kaggle/working/drive/\n","\n","\n","\n","## download tool & system updates\n","!echo \"updating system......\"\n","!apt update -y\n","!apt dist-upgrade -y\n","clear_output()\n","\n","\n","## webui\n","!echo \"installing webui......\"\n","!git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git /kaggle/working/webui/\n","%cd /kaggle/working/webui/\n","!git pull\n","%pip install -r requirements.txt\n","\n","\n","## extensions for webui\n","extensions = {\"https://github.com/AUTOMATIC1111/stable-diffusion-webui-aesthetic-gradients.git\": \"aesthetic-gradients\",\n","              \"https://github.com/yfszzx/stable-diffusion-webui-images-browser.git\": \"images-browser\",\n","              \"https://github.com/deforum-art/deforum-for-automatic1111-webui.git\": \"deforum\",\n","              \"https://github.com/7eu7d7/DreamArtist-sd-webui-extension.git\": \"dreamartist\",\n","              \"https://github.com/ThereforeGames/unprompted.git\": \"unprompted\",\n","              \"https://github.com/animerl/novelai-2-local-prompt.git\": \"nai-prompt-converter\",\n","              \"https://github.com/dtlnor/stable-diffusion-webui-localization-zh_CN.git\": \"localization-zhcn\",\n","              \"https://github.com/DominikDoom/a1111-sd-webui-tagcomplete.git\": \"tagcomplete\"\n","              }\n","for repo,name in extensions.items():\n","  if name not in disabled_extensions:\n","    !echo \"installing extension {name}......\"\n","    !git clone {repo} /kaggle/working/webui/extensions/{name}/\n","    %cd /kaggle/working/webui/extensions/{name}/\n","    !git pull\n","\n","\n","## xformers\n","if xformers == \"from_source\":\n","  !echo \"installing xformers......\"\n","  !git clone --recursive https://github.com/facebookresearch/xformers.git /kaggle/working/buildspace/xformers/\n","  %cd /kaggle/working/buildspace/xformers/\n","  %pip install -r requirements.txt\n","  %pip install -e .\n","elif xformers == \"from_pypi\":\n","  !echo \"installing xformers......\"\n","  %pip install torch torchvision torchaudio pytorch-lightning\n","  %pip install -U xformers\n","elif xformers == \"none\":\n","  pass\n","\n","\n","## models\n","!mkdir -p /kaggle/working/webui/models/Stable-diffusion/ /kaggle/working/webui/models/hypernetworks/ /kaggle/working/webui/embeddings/\n","%cd /kaggle/working/webui/models/Stable-diffusion/\n","\n","# official stable diffusion\n","if official == True:\n","  !echo \"downloading official diffusion......\"\n","  !curl -Lo official.ckpt https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.ckpt\n","\n","# anything\n","if anything == True:\n","  !echo \"downloading anything......\"\n","  #!curl -Lo anything.ckpt https://huggingface.co/xyn-ai/anything-v4.0/resolve/main/anything-v4.5-pruned-fp32.ckpt # old version, will be using the newer one for now\n","  !curl -Lo anything.safetensors https://civitai.com/api/download/models/90854\n","\n","# novel ai\n","if nai == True:\n","  !echo \"downloading novel ai......\"\n","  !curl -Lo nai.ckpt https://huggingface.co/acheong08/NovelAI/resolve/main/stableckpt/animefull-final-pruned/model.ckpt\n","  !curl -Lo nai.yaml https://huggingface.co/acheong08/NovelAI/resolve/main/stableckpt/animefull-final-pruned/config.yaml\n","  !curl -Lo nai.vae.pt https://huggingface.co/acheong08/NovelAI/resolve/main/stableckpt/animevae.pt\n","\n","# waifu diffusion\n","if waifu == True:\n","  !echo \"downloading waifu diffusion......\"\n","  !curl -Lo waifu.ckpt https://huggingface.co/waifu-diffusion/wd-1-5-beta3/resolve/main/wd-beta3-base-fp16.safetensors\n","\n","# counterfait\n","if counterfait == True:\n","  !echo \"downloading counterfait......\"\n","  !curl -Lo counterfait.safetensors https://huggingface.co/gsdf/Counterfeit-V3.0/resolve/main/Counterfeit-V3.0_fp32.safetensors\n","\n","\n","# embeddings - negative prompt\n","!curl -Lo /kaggle/working/webui/embeddings/easynegative.safetensors https://huggingface.co/datasets/gsdf/EasyNegative/resolve/main/EasyNegative.safetensors\n","\n","\n","## server tunneling tool\n","!echo \"downloading server tunneling tool......\"\n","!curl -Ls https://github.com/ekzhang/bore/releases/download/v0.5.0/bore-v0.5.0-x86_64-unknown-linux-musl.tar.gz | tar zx -C /kaggle/working/tools/ && chmod +x /kaggle/working/tools/bore\n","!curl -Lo /kaggle/working/tools/cloudflared https://github.com/cloudflare/cloudflared/releases/latest/download/cloudflared-linux-amd64 && chmod +x /kaggle/working/tools/cloudflared"]},{"cell_type":"code","execution_count":null,"metadata":{"id":"2Hp-zFycUmG4","trusted":true},"outputs":[],"source":["#@title ### 3. Server\n","\n","%cd /kaggle/working/webui/\n","if oHNo_My_SErVEr_keEpS_cRaShiNg_iTSeLF_HeLP_mE == False:\n","  !python launch.py --medvram {additional_server_parameters} & /kaggle/working/tools/bore local 7860 --to bore.pub & /kaggle/working/tools/cloudflared tunnel --url localhost:7860\n","else:\n","  !python launch.py --lowvram {additional_server_parameters} & /kaggle/working/tools/bore local 7860 --to bore.pub & /kaggle/working/tools/cloudflared tunnel --url localhost:7860"]}],"metadata":{"accelerator":"GPU","colab":{"gpuType":"T4","private_outputs":true,"provenance":[]},"kernelspec":{"display_name":"Python 3","language":"python","name":"python3"},"language_info":{"codemirror_mode":{"name":"ipython","version":3},"file_extension":".py","mimetype":"text/x-python","name":"python","nbconvert_exporter":"python","pygments_lexer":"ipython3","version":"3.11.3"}},"nbformat":4,"nbformat_minor":4}