Stable-DiffusionENG.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "provenance": []
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    },
    "language_info": {
      "name": "python"
    },
    "accelerator": "GPU",
    "gpuClass": "standard"
  },
  "cells": [
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "id": "uKc4Nj4AXhMQ",
        "cellView": "form"
      },
      "outputs": [],
      "source": [
        "#@title Stable Diffusion web-ui colab by [Meldoner](https://github.com/Meldoner)\n",
        "%cd /content\n",
        "!git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui\n",
        "!git clone https://github.com/yfszzx/stable-diffusion-webui-images-browser /content/stable-diffusion-webui/extensions/stable-diffusion-webui-images-browser\n",
        "!wget -nc --content-disposition https://huggingface.co/datasets/gsdf/EasyNegative/resolve/main/EasyNegative.safetensors -P /content/stable-diffusion-webui/embeddings\n",
        "#@markdown **Choose a ready-made model**\n",
        "model = \"openjourneyv4\" #@param [\"openjourneyv4\",\"sd2.1\",\"sd1.5\",\"abyss-orange-mix3\",\"anythingv3\",\"anythingv4.5\",\"deliberate2\",\"dreamlike\",\"dreamshaper5\",\"kenshi\",\"protogenx5.8\",\"hassaku\",\"realistic-visionv2\",\"corneos-7th-heaven-mix\",\"pastel-mix\",\"counterfeit-v2.5\",\"rev-animated\"]\n",
        "#@markdown **IMPORTANT**: some models require **VAE**, choose it in the settings\\\n",
        "#@markdown By default installed [EasyNegative](https://civitai.com/models/7808/easynegative)\n",
        "\n",
        "#@markdown **Click if you want to use your model**\n",
        "load_own_model = False #@param{type:\"boolean\"}\n",
        "model_url = \"Paste the link to the model (ckpt/safetensors)\" #@param {type:\"string\"}\n",
        "\n",
        "#@markdown **Click if you want to use your VAE**\n",
        "load_own_vae = False #@param{type:\"boolean\"}\n",
        "vae_url = \"Paste the link to the VAE\" #@param {type:\"string\"}\n",
        "\n",
        "#@markdown **Click if you want to use your embeddings**\n",
        "load_own_embedding = False #@param{type:\"boolean\"}\n",
        "embedding_url = \"Paste the link to the embedding\" #@param {type:\"string\"}\n",
        "\n",
        "#@markdown **Click if you want to use your LORA**\n",
        "load_own_lora = False #@param{type:\"boolean\"}\n",
        "lora_url = \"Paste the link to the LORA\" #@param {type:\"string\"}\n",
        "\n",
        "#@markdown You can load multiple models(and other stuff) by simply restarting the script\n",
        "\n",
        "#@markdown **If you want me to add more models, just create a request [here](https://github.com/Meldoner/stable-diffusion-webui-colab/issues)**\n",
        "model_urls = {\n",
        "    'openjourneyv4': 'https://huggingface.co/prompthero/openjourney-v4/resolve/main/openjourney-v4.ckpt',\n",
        "    'abyss-orange-mix3': ['https://huggingface.co/WarriorMama777/OrangeMixs/resolve/main/Models/AbyssOrangeMix3/AOM3_orangemixs.safetensors', 'https://huggingface.co/WarriorMama777/OrangeMixs/resolve/main/VAEs/orangemix.vae.pt'],\n",
        "    'anythingv3': ['https://huggingface.co/admruul/anything-v3.0/resolve/main/Anything-V3.0-pruned.safetensors', 'https://huggingface.co/admruul/anything-v3.0/resolve/main/Anything-V3.0.vae.pt'],\n",
        "    'anythingv4.5': ['https://huggingface.co/andite/anything-v4.0/resolve/main/anything-v4.5-pruned.safetensors', 'https://huggingface.co/andite/anything-v4.0/resolve/main/anything-v4.0.vae.pt'],\n",
        "    'deliberate2': 'https://huggingface.co/XpucT/Deliberate/resolve/main/Deliberate_v2.safetensors',\n",
        "    'dreamlike': 'https://huggingface.co/dreamlike-art/dreamlike-diffusion-1.0/resolve/main/dreamlike-diffusion-1.0.safetensors',\n",
        "    'dreamshaper5': 'https://huggingface.co/Lykon/DreamShaper/resolve/main/DreamShaper_5_beta2_BakedVae_fp16.safetensors',\n",
        "    'kenshi': 'https://huggingface.co/SweetLuna/Kenshi/resolve/main/KENSHI%2001/KENSHI01_baked.ckpt',\n",
        "    'protogenx5.8': 'https://huggingface.co/darkstorm2150/Protogen_x5.8_Official_Release/resolve/main/ProtoGen_X5.8.safetensors',\n",
        "    'sd2.1': 'https://huggingface.co/stabilityai/stable-diffusion-2-1/resolve/main/v2-1_768-ema-pruned.safetensors',\n",
        "    'sd1.5': 'https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.safetensors',\n",
        "    'hassaku': ['https://huggingface.co/sinkinai/hassaku-hentai/resolve/main/hassakuHentaiModel_v11.safetensors', 'https://huggingface.co/WarriorMama777/OrangeMixs/resolve/main/VAEs/orangemix.vae.pt'],\n",
        "    'realistic-visionv2': ['https://huggingface.co/SG161222/Realistic_Vision_V2.0/resolve/main/Realistic_Vision_V2.0.safetensors', 'https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.safetensors'],\n",
        "    'corneos-7th-heaven-mix': ['https://huggingface.co/dwarfbum/corneos7thHeavenMix_v2/resolve/main/corneos7thHeavenMix_v2.safetensors', 'https://huggingface.co/andite/anything-v4.0/resolve/main/anything-v4.0.vae.pt'],\n",
        "    'pastel-mix': 'https://huggingface.co/andite/pastel-mix/resolve/main/pastelmix.safetensors',\n",
        "    'counterfeit-v2.5': ['https://huggingface.co/gsdf/Counterfeit-V2.5/resolve/main/Counterfeit-V2.5_pruned.safetensors','https://huggingface.co/gsdf/Counterfeit-V2.5/resolve/main/Counterfeit-V2.5.vae.pt'],\n",
        "    'rev-animated': ['https://huggingface.co/s6yx/ReV_Animated/resolve/main/rev_1.2.2/rev_1.2.2.safetensors','https://huggingface.co/WarriorMama777/OrangeMixs/resolve/main/VAEs/orangemix.vae.pt']\n",
        "}\n",
        "\n",
        "if model in model_urls and not load_own_model:\n",
        "    if isinstance(model_urls[model], list):\n",
        "          !wget -nc --content-disposition {model_urls[model][0]} -P /content/stable-diffusion-webui/models/Stable-diffusion\n",
        "          !wget -nc --content-disposition {model_urls[model][1]} -P /content/stable-diffusion-webui/models/VAE\n",
        "    else:\n",
        "        !wget -nc --content-disposition {model_urls[model]} -P /content/stable-diffusion-webui/models/Stable-diffusion\n",
        "\n",
        "if load_own_model:\n",
        "  !wget -nc --content-disposition {model_url} -P /content/stable-diffusion-webui/models/Stable-diffusion\n",
        "\n",
        "if load_own_vae:\n",
        "  !wget -nc --content-disposition {vae_url} -P /content/stable-diffusion-webui/models/VAE\n",
        "\n",
        "if load_own_embedding:\n",
        "  !wget -nc --content-disposition {embedding_url} -P /content/stable-diffusion-webui/embeddings\n",
        "\n",
        "if load_own_lora:\n",
        "  !wget -nc --content-disposition {lora_url} -P /content/stable-diffusion-webui/models/Lora\n",
        "\n",
        "!pip install -q fastapi==0.90.1 -U\n",
        "!pip install -q torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1+cu118 torchtext==0.15.1 torchdata==0.6.0 --extra-index-url https://download.pytorch.org/whl/cu118 -U\n",
        "!pip install -q xformers==0.0.18 triton==2.0.0 -U\n",
        "%cd stable-diffusion-webui\n",
        "!COMMANDLINE_ARGS=\"--share --disable-safe-unpickle --no-half-vae --xformers --gradio-queue --enable-insecure-extension-access\" REQS_FILE=\"requirements.txt\" python launch.py"
      ]
    }
  ]
}