Stable_Diffusion_WebUi_Altryne.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "jnXUwtACCHOy"
   },
   "source": [
    "# Welcome to Stable Diffusion WebUI 1.4! by [@altryne](https://twitter.com/altryne/) |  [![ko-fi](https://ko-fi.com/img/githubbutton_sm.svg)](https://ko-fi.com/N4N3DWMR1) \n",
    "\n",
    "This colab runs the latest webui version from the repo https://github.com/hlky/stable-diffusion-webui\n",
    "\n",
    "---\n",
    "\n",
    "If this colab helped you, support me on ko-fi and don't forget to subscribe to my awesome list https://github.com/altryne/awesome-ai-art-image-synthesis\n"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "uKA724szI_aG"
   },
   "source": []
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "4eodrBxeCgOW"
   },
   "source": [
    "## 1 -  Setup stage"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "YiG8fBPrI99p"
   },
   "source": [
    "### 1.0"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {
    "id": "cAsdhafsBvEr"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "GPU 0: Quadro M4000 (UUID: GPU-e8c5ba2e-8d87-cf7d-a4f0-d4e9d1cedeb4)\n"
     ]
    }
   ],
   "source": [
    "!nvidia-smi -L"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "RXasLtrYC9uj"
   },
   "source": [
    "### 1.1 Download repo and install\n",
    "\n",
    "Clone git repo and setup miniconda\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {
    "cellView": "form",
    "id": "LSCxESSsDGVh"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "/content\n",
      "Cloning into 'stable-diffusion'...\n",
      "remote: Enumerating objects: 1162, done.\u001b[K\n",
      "remote: Total 1162 (delta 0), reused 0 (delta 0), pack-reused 1162\u001b[K\n",
      "Receiving objects: 100% (1162/1162), 42.84 MiB | 46.08 MiB/s, done.\n",
      "Resolving deltas: 100% (564/564), done.\n",
      "/content/stable-diffusion\n",
      "--2022-09-10 21:23:48--  https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh\n",
      "Resolving repo.anaconda.com (repo.anaconda.com)... 104.16.130.3, 104.16.131.3, 2606:4700::6810:8203, ...\n",
      "Connecting to repo.anaconda.com (repo.anaconda.com)|104.16.130.3|:443... connected.\n",
      "HTTP request sent, awaiting response... 200 OK\n",
      "Length: 76607678 (73M) [application/x-sh]\n",
      "Saving to: ‘Miniconda3-latest-Linux-x86_64.sh’\n",
      "\n",
      "Miniconda3-latest-L 100%[===================>]  73.06M   143MB/s    in 0.5s    \n",
      "\n",
      "2022-09-10 21:23:48 (143 MB/s) - ‘Miniconda3-latest-Linux-x86_64.sh’ saved [76607678/76607678]\n",
      "\n",
      "PREFIX=/usr/local\n",
      "Unpacking payload ...\n",
      "Collecting package metadata (current_repodata.json): -                        done\n",
      "Solving environment: done\n",
      "\n",
      "# All requested packages already installed.\n",
      "\n",
      "installation finished.\n"
     ]
    }
   ],
   "source": [
    "#@markdown ## Download the stable-diffusion webui repo\n",
    "#@markdown And install colab related conda (uncheck stable version if you like new features)\n",
    "%mkdir /content\n",
    "%cd /content\n",
    "!git clone https://github.com/hlky/stable-diffusion\n",
    "%cd /content/stable-diffusion\n",
    "stable_version = False #@param {type:\"boolean\"}\n",
    "if stable_version: \n",
    "  !git checkout colab-pin\n",
    "\n",
    "import sys\n",
    "!wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh\n",
    "!chmod +x Miniconda3-latest-Linux-x86_64.sh\n",
    "!bash ./Miniconda3-latest-Linux-x86_64.sh -b -f -p /usr/local\n",
    "sys.path.append('/usr/local/lib/python3.7/site-packages/')\n",
    "!rm Miniconda3-latest-Linux-x86_64.sh"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "MVJzvQGdDn6C"
   },
   "source": [
    "### 1.2 Environment setup\n",
    "Setup environment, Gfpgan and Real-ESRGAN. Takes about 5-6 minutes"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {
    "cellView": "form",
    "id": "PvH2zyoiD0q8"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\n",
      "SpecNotFound: Invalid name, try the format: user/package\n",
      "\n"
     ]
    }
   ],
   "source": [
    "#@markdown ### Set up conda environment - Takes a while\n",
    "!conda env update -n base -f /content/stable-diffusion/environment.yaml "
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "vhZQqFGxEHcA"
   },
   "source": [
    "### 1.3 Setup Upscalers - CFPGan and ESRGAN"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "cellView": "form",
    "id": "M3o7slcTEQvp"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[Errno 2] No such file or directory: '/content/stable-diffusion/src/gfpgan/'\n",
      "/content/stable-diffusion\n",
      "Collecting basicsr\n",
      "  Downloading basicsr-1.4.2.tar.gz (172 kB)\n",
      "\u001b[K     |████████████████████████████████| 172 kB 22.5 MB/s eta 0:00:01\n",
      "^C\n",
      "\u001b[31mERROR: Operation cancelled by user\u001b[0m\n",
      "python: can't open file '/content/stable-diffusion/setup.py': [Errno 2] No such file or directory\n",
      "Collecting realesrgan\n",
      "  Downloading realesrgan-0.2.5.0-py3-none-any.whl (25 kB)\n",
      "Collecting basicsr>=1.3.3.11\n",
      "  Using cached basicsr-1.4.2.tar.gz (172 kB)\n",
      "Collecting opencv-python\n",
      "  Downloading opencv_python-4.6.0.66-cp36-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (60.9 MB)\n",
      "\u001b[K     |████████████████████████████████| 60.9 MB 230 kB/s eta 0:00:01     |█████████████████████           | 40.1 MB 6.1 MB/s eta 0:00:04\n",
      "\u001b[?25hCollecting facexlib>=0.2.0.3\n",
      "  Downloading facexlib-0.2.5-py3-none-any.whl (59 kB)\n",
      "\u001b[K     |████████████████████████████████| 59 kB 13.5 MB/s eta 0:00:01\n",
      "\u001b[?25hCollecting torch>=1.7\n",
      "  Using cached torch-1.12.1-cp39-cp39-manylinux1_x86_64.whl (776.4 MB)\n",
      "Requirement already satisfied: tqdm in /usr/local/lib/python3.9/site-packages (from realesrgan) (4.63.0)\n",
      "Collecting gfpgan>=0.2.1\n",
      "  Downloading gfpgan-1.3.5-py3-none-any.whl (47 kB)\n",
      "\u001b[K     |████████████████████████████████| 47 kB 8.3 MB/s  eta 0:00:01\n",
      "\u001b[?25hCollecting torchvision\n",
      "  Downloading torchvision-0.13.1-cp39-cp39-manylinux1_x86_64.whl (19.1 MB)\n",
      "\u001b[K     |████████████████████████████████| 19.1 MB 22.0 MB/s eta 0:00:01\n",
      "\u001b[?25hCollecting Pillow\n",
      "  Downloading Pillow-9.2.0-cp39-cp39-manylinux_2_28_x86_64.whl (3.2 MB)\n",
      "\u001b[K     |████████████████████████████████| 3.2 MB 40.3 MB/s eta 0:00:01\n",
      "\u001b[?25hCollecting numpy\n",
      "  Using cached numpy-1.23.3-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (17.1 MB)\n",
      "Collecting addict\n",
      "  Downloading addict-2.4.0-py3-none-any.whl (3.8 kB)\n",
      "Collecting future\n",
      "  Downloading future-0.18.2.tar.gz (829 kB)\n",
      "\u001b[K     |████████████████████████████████| 829 kB 74.4 MB/s eta 0:00:01\n",
      "\u001b[?25hCollecting lmdb\n",
      "  Downloading lmdb-1.3.0-cp39-cp39-manylinux_2_12_x86_64.manylinux2010_x86_64.whl (305 kB)\n",
      "\u001b[K     |████████████████████████████████| 305 kB 54.5 MB/s eta 0:00:01\n",
      "\u001b[?25hCollecting pyyaml\n",
      "  Downloading PyYAML-6.0-cp39-cp39-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_12_x86_64.manylinux2010_x86_64.whl (661 kB)\n",
      "\u001b[K     |████████████████████████████████| 661 kB 53.8 MB/s eta 0:00:01\n",
      "\u001b[?25hRequirement already satisfied: requests in /usr/local/lib/python3.9/site-packages (from basicsr>=1.3.3.11->realesrgan) (2.27.1)\n",
      "Collecting scikit-image\n",
      "  Downloading scikit_image-0.19.3-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (14.0 MB)\n",
      "\u001b[K     |████████████████████████████████| 14.0 MB 35.0 MB/s eta 0:00:01\n",
      "\u001b[?25hCollecting scipy\n",
      "  Downloading scipy-1.9.1-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (43.9 MB)\n",
      "\u001b[K     |████████████████████████████████| 43.9 MB 101 kB/s eta 0:00:01\n",
      "\u001b[?25hCollecting tb-nightly\n",
      "  Downloading tb_nightly-2.11.0a20220910-py3-none-any.whl (5.9 MB)\n",
      "\u001b[K     |████████████████████████████████| 5.9 MB 26.0 MB/s eta 0:00:01\n",
      "\u001b[?25hCollecting yapf\n",
      "  Downloading yapf-0.32.0-py2.py3-none-any.whl (190 kB)\n",
      "\u001b[K     |████████████████████████████████| 190 kB 61.6 MB/s eta 0:00:01\n",
      "\u001b[?25hCollecting filterpy\n",
      "  Downloading filterpy-1.4.5.zip (177 kB)\n",
      "\u001b[K     |████████████████████████████████| 177 kB 47.3 MB/s eta 0:00:01\n",
      "^C\n",
      "\u001b[31mERROR: Operation cancelled by user\u001b[0m\n",
      "--2022-09-10 21:27:47--  https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth\n",
      "Resolving github.com (github.com)... 140.82.113.4\n",
      "Connecting to github.com (github.com)|140.82.113.4|:443... connected.\n",
      "HTTP request sent, awaiting response... 302 Found\n",
      "Location: https://objects.githubusercontent.com/github-production-release-asset-2e65be/349321229/e9847322-b8b1-4ec2-9620-5146eb8a9e4b?X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Credential=AKIAIWNJYAX4CSVEH53A%2F20220910%2Fus-east-1%2Fs3%2Faws4_request&X-Amz-Date=20220910T212748Z&X-Amz-Expires=300&X-Amz-Signature=63be5c765832504bc3e2892dcdd97d1f771f63427c9bf747858c11239bfa983e&X-Amz-SignedHeaders=host&actor_id=0&key_id=0&repo_id=349321229&response-content-disposition=attachment%3B%20filename%3DGFPGANv1.3.pth&response-content-type=application%2Foctet-stream [following]\n",
      "--2022-09-10 21:27:48--  https://objects.githubusercontent.com/github-production-release-asset-2e65be/349321229/e9847322-b8b1-4ec2-9620-5146eb8a9e4b?X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Credential=AKIAIWNJYAX4CSVEH53A%2F20220910%2Fus-east-1%2Fs3%2Faws4_request&X-Amz-Date=20220910T212748Z&X-Amz-Expires=300&X-Amz-Signature=63be5c765832504bc3e2892dcdd97d1f771f63427c9bf747858c11239bfa983e&X-Amz-SignedHeaders=host&actor_id=0&key_id=0&repo_id=349321229&response-content-disposition=attachment%3B%20filename%3DGFPGANv1.3.pth&response-content-type=application%2Foctet-stream\n",
      "Resolving objects.githubusercontent.com (objects.githubusercontent.com)... 185.199.108.133, 185.199.110.133, 185.199.111.133, ...\n",
      "Connecting to objects.githubusercontent.com (objects.githubusercontent.com)|185.199.108.133|:443... connected.\n",
      "HTTP request sent, awaiting response... 200 OK\n",
      "Length: 348632874 (332M) [application/octet-stream]\n",
      "Saving to: ‘experiments/pretrained_models/GFPGANv1.3.pth’\n",
      "\n",
      "GFPGANv1.3.pth      100%[===================>] 332.48M   104MB/s    in 3.2s    \n",
      "\n",
      "2022-09-10 21:27:51 (104 MB/s) - ‘experiments/pretrained_models/GFPGANv1.3.pth’ saved [348632874/348632874]\n",
      "\n",
      "[Errno 2] No such file or directory: '/content/stable-diffusion/src/realesrgan/'\n",
      "/content/stable-diffusion\n",
      "--2022-09-10 21:27:52--  https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth\n",
      "Resolving github.com (github.com)... ^C\n",
      "--2022-09-10 21:27:55--  https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.4/RealESRGAN_x4plus_anime_6B.pth\n",
      "Resolving github.com (github.com)... 140.82.113.4\n",
      "Connecting to github.com (github.com)|140.82.113.4|:443... connected.\n",
      "HTTP request sent, awaiting response... 302 Found\n",
      "Location: https://objects.githubusercontent.com/github-production-release-asset-2e65be/387326890/4f59d7c8-d03f-494e-8595-ae23af075393?X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Credential=AKIAIWNJYAX4CSVEH53A%2F20220910%2Fus-east-1%2Fs3%2Faws4_request&X-Amz-Date=20220910T212745Z&X-Amz-Expires=300&X-Amz-Signature=731dd67fbbd922ab61a7e597d03aac0c9b046c1f151a678441ceee5f59b6147b&X-Amz-SignedHeaders=host&actor_id=0&key_id=0&repo_id=387326890&response-content-disposition=attachment%3B%20filename%3DRealESRGAN_x4plus_anime_6B.pth&response-content-type=application%2Foctet-stream [following]\n",
      "--2022-09-10 21:27:55--  https://objects.githubusercontent.com/github-production-release-asset-2e65be/387326890/4f59d7c8-d03f-494e-8595-ae23af075393?X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Credential=AKIAIWNJYAX4CSVEH53A%2F20220910%2Fus-east-1%2Fs3%2Faws4_request&X-Amz-Date=20220910T212745Z&X-Amz-Expires=300&X-Amz-Signature=731dd67fbbd922ab61a7e597d03aac0c9b046c1f151a678441ceee5f59b6147b&X-Amz-SignedHeaders=host&actor_id=0&key_id=0&repo_id=387326890&response-content-disposition=attachment%3B%20filename%3DRealESRGAN_x4plus_anime_6B.pth&response-content-type=application%2Foctet-stream\n",
      "Resolving objects.githubusercontent.com (objects.githubusercontent.com)... 185.199.111.133, 185.199.110.133, 185.199.109.133, ...\n",
      "Connecting to objects.githubusercontent.com (objects.githubusercontent.com)|185.199.111.133|:443... connected.\n",
      "HTTP request sent, awaiting response... 200 OK\n",
      "Length: 17938799 (17M) [application/octet-stream]\n",
      "Saving to: ‘experiments/pretrained_models/RealESRGAN_x4plus_anime_6B.pth’\n",
      "\n",
      "RealESRGAN_x4plus_a 100%[===================>]  17.11M  --.-KB/s    in 0.1s    \n",
      "\n",
      "2022-09-10 21:27:55 (150 MB/s) - ‘experiments/pretrained_models/RealESRGAN_x4plus_anime_6B.pth’ saved [17938799/17938799]\n",
      "\n",
      "/content/stable-diffusion\n",
      "--2022-09-10 21:27:56--  https://github.com/matomo-org/travis-scripts/blob/master/fonts/Arial.ttf?raw=true\n",
      "Resolving github.com (github.com)... 140.82.112.3\n",
      "Connecting to github.com (github.com)|140.82.112.3|:443... connected.\n",
      "HTTP request sent, awaiting response... 302 Found\n",
      "Location: https://github.com/matomo-org/travis-scripts/raw/master/fonts/Arial.ttf [following]\n",
      "--2022-09-10 21:27:56--  https://github.com/matomo-org/travis-scripts/raw/master/fonts/Arial.ttf\n",
      "Reusing existing connection to github.com:443.\n",
      "HTTP request sent, awaiting response... 302 Found\n",
      "Location: https://raw.githubusercontent.com/matomo-org/travis-scripts/master/fonts/Arial.ttf [following]\n",
      "--2022-09-10 21:27:56--  https://raw.githubusercontent.com/matomo-org/travis-scripts/master/fonts/Arial.ttf\n",
      "Resolving raw.githubusercontent.com (raw.githubusercontent.com)... 185.199.111.133, 185.199.108.133, 185.199.109.133, ...\n",
      "Connecting to raw.githubusercontent.com (raw.githubusercontent.com)|185.199.111.133|:443... connected.\n",
      "HTTP request sent, awaiting response... 200 OK\n",
      "Length: 275572 (269K) [application/octet-stream]\n",
      "Saving to: ‘arial.ttf’\n",
      "\n",
      "arial.ttf           100%[===================>] 269.11K  --.-KB/s    in 0.007s  \n",
      "\n",
      "2022-09-10 21:27:56 (38.2 MB/s) - ‘arial.ttf’ saved [275572/275572]\n",
      "\n"
     ]
    }
   ],
   "source": [
    "\n",
    "#@markdown ### Build upscalers support\n",
    "#@markdown **GFPGAN** Automatically correct distorted faces with a built-in GFPGAN option, fixes them in less than half a second\n",
    "#@markdown **ESRGAN** Boosts the resolution of images with a built-in RealESRGAN option\n",
    "#@markdown LDSR and GoBig enable amazing upscale options in the new Image Lab\n",
    "\n",
    "add_CFP = True #@param {type:\"boolean\"} \n",
    "add_ESR = True #@param {type:\"boolean\"}\n",
    "add_LDSR = False #@param {type:\"boolean\"} \n",
    "#@markdown ⚠️ LDSR is 1.9GB and make take time to download\n",
    "\n",
    "if add_CFP:\n",
    "  %cd /content/stable-diffusion/src/gfpgan/\n",
    "  !pip install basicsr facexlib yapf lmdb opencv-python pyyaml tb-nightly --no-deps\n",
    "  !python setup.py develop\n",
    "  !pip install realesrgan\n",
    "  !wget https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth -P experiments/pretrained_models\n",
    "if add_ESR:\n",
    "  %cd /content/stable-diffusion/src/realesrgan/\n",
    "  !wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth -P experiments/pretrained_models\n",
    "  !wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.4/RealESRGAN_x4plus_anime_6B.pth -P experiments/pretrained_models\n",
    "if add_LDSR:\n",
    "  %cd /content/stable-diffusion/src\n",
    "  !git clone https://github.com/devilismyfriend/latent-diffusion\n",
    "  %cd latent-diffusion\n",
    "  %mkdir -p experiments/\n",
    "  %cd experiments/\n",
    "  %mkdir -p pretrained_models\n",
    "  %cd pretrained_models\n",
    "  #project.yaml download\n",
    "  !wget -O project.yaml https://heibox.uni-heidelberg.de/f/31a76b13ea27482981b4/?dl=1\n",
    "  #model.ckpt model download\n",
    "  !wget -O model.ckpt https://heibox.uni-heidelberg.de/f/578df07c8fc04ffbadf3/?dl=1\n",
    "\n",
    "%cd /content/stable-diffusion/\n",
    "!wget https://github.com/matomo-org/travis-scripts/blob/master/fonts/Arial.ttf?raw=true -O arial.ttf"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "mBFOSYR91AeT"
   },
   "source": [
    "### 1.4 Connect to Google Drive"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "cellView": "form",
    "id": "vkgBVo5OEpqn"
   },
   "outputs": [],
   "source": [
    "#@markdown # Load the stable-diffusion model\n",
    "\n",
    "#@markdown **Model Path Variables**\n",
    "# ask for the link\n",
    "print(\"Local Path Variables:\\n\")\n",
    "\n",
    "models_path = \"/content/models\" #@param {type:\"string\"}\n",
    "output_path = \"/content/output\" #@param {type:\"string\"}\n",
    "\n",
    "#@markdown **Download the model if it isn't already in the 'models_path' folder (Optional)**\n",
    "\n",
    "#@markdown To download the model, you need to have accepted the terms [HERE](https://huggingface.co/CompVis/stable-diffusion-v1-4)\n",
    "#@markdown and have copied a token from [HERE](https://huggingface.co/settings/tokens)\n",
    "download_if_missing = True #@param {type:\"boolean\"}\n",
    "token = \"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\" #@param {type:\"string\"}\n",
    "\n",
    "#@markdown **Google Drive Path Variables (Optional)**\n",
    "mount_google_drive = False #@param {type:\"boolean\"}\n",
    "force_remount = False\n",
    "\n",
    "%cd /content/\n",
    "import os\n",
    "mount_success = True\n",
    "if mount_google_drive:\n",
    "    from google.colab import drive\n",
    "    try:\n",
    "        drive_path = \"/content/drive\"\n",
    "        drive.mount(drive_path,force_remount=force_remount)\n",
    "        models_path_gdrive = \"/content/drive/MyDrive/AI/models\" #@param {type:\"string\"}\n",
    "        output_path_gdrive = \"/content/drive/MyDrive/AI/StableDiffusion\" #@param {type:\"string\"}\n",
    "        models_path = models_path_gdrive\n",
    "        output_path = output_path_gdrive\n",
    "    except:\n",
    "        print(\"...error mounting drive or with drive path variables\")\n",
    "        print(\"...reverting to default path variables\")\n",
    "        mount_success = False\n",
    "\n",
    "os.makedirs(models_path, exist_ok=True)\n",
    "os.makedirs(output_path, exist_ok=True)\n",
    "\n",
    "if download_if_missing:\n",
    "    if not mount_success:\n",
    "        print(\"Downloading model to \" + models_path + \" due to gdrive mount error\")\n",
    "    if token == \"\":\n",
    "        print(\"No token provided. Assuming model is already in \" + models_path)\n",
    "    elif not os.path.exists(models_path + '/sd-v1-4.ckpt'):\n",
    "        !git lfs install --system --skip-repo\n",
    "        !mkdir sd-model\n",
    "        %cd /content/sd-model/\n",
    "        !git init\n",
    "        !git remote add -f origin \"https://USER:{token}@huggingface.co/CompVis/stable-diffusion-v-1-4-original\"\n",
    "        !git config core.sparsecheckout true\n",
    "        !echo \"sd-v1-4.ckpt\" > .git/info/sparse-checkout\n",
    "        !git pull origin main\n",
    "        !mv '/content/sd-model/sd-v1-4.ckpt' '{models_path}/'\n",
    "    else:\n",
    "        print(\"Model already downloaded, moving to next step\")\n",
    "\n",
    "print(f\"models_path: {models_path}\")\n",
    "print(f\"output_path: {output_path}\")"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "Wt_tzv_eJ9Qz"
   },
   "source": [
    "## 2 - Run the Stable Diffusion webui"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "ip0o1IqmQNON"
   },
   "source": [
    "### 2.1 Optional - Set webUI settings and configs before running "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "cellView": "form",
    "id": "3t1DaxIfRmSm"
   },
   "outputs": [],
   "source": [
    "#@markdown # Launch preferences - Advanced\n",
    "share_password=\"\" #@param {type:\"string\"}\n",
    "#@markdown * Add a password to your webui\n",
    "defaults=\"configs/webui/webui.yaml\" #@param {type:\"string\"}\n",
    "#@markdown * path to configuration file providing UI defaults, uses same format as cli parameter)  \n",
    "#@markdown Edit this file if you want to change the default settings UI launches with\n",
    "\n",
    "#@markdown ---\n",
    "save_metadata = False #@param {type:\"boolean\"}\n",
    "#@markdown * Whether to embed the generation parameters in the sample images\n",
    "skip_grid = False #@param {type:\"boolean\"}\n",
    "#@markdown * Do not save a grid, only individual samples. Helpful when evaluating lots of samples\n",
    "skip_save = False #@param {type:\"boolean\"}\n",
    "#@markdown * Do not save individual samples as files. For speed measurements\n",
    "optimized = False #@param {type:\"boolean\"}\n",
    "#@markdown * Load the model onto the device piecemeal instead of all at once to reduce VRAM usage at the cost of performance\n",
    "optimized_turbo = True #@param {type:\"boolean\"}\n",
    "#@markdown * Alternative optimization mode that does not save as much VRAM but runs siginificantly faster\n",
    "no_verify_input = False #@param {type:\"boolean\"}\n",
    "#@markdown * Do not verify input to check if it's too long\n",
    "no_half = False #@param {type:\"boolean\"}\n",
    "#@markdown * Do not switch the model to 16-bit floats\n",
    "no_progressbar_hiding = True #@param {type:\"boolean\"}\n",
    "#@markdown * Do not hide progressbar in gradio UI\n",
    "extra_models_cpu = False #@param {type:\"boolean\"}\n",
    "#@markdown * Run extra models (GFGPAN/ESRGAN) on cpu\n",
    "esrgan_cpu = True #@param {type:\"boolean\"}\n",
    "#@markdown * run ESRGAN on cpu\n",
    "gfpgan_cpu = False #@param {type:\"boolean\"}\n",
    "#@markdown * run GFPGAN on cpu\n",
    "\n",
    "\n",
    "run_string_with_variables = {\n",
    " '--save-metadata': f'{save_metadata}',\n",
    " '--skip-grid': f'{skip_grid}',\n",
    " '--skip-save': f'{skip_save}',\n",
    " '--optimized': f'{optimized}',\n",
    " '--optimized-turbo': f'{optimized_turbo}',\n",
    " '--no-verify-input': f'{no_verify_input}',\n",
    " '--no-half': f'{no_half}',\n",
    " '--no-progressbar-hiding': f'{no_progressbar_hiding}',\n",
    " '--extra-models-cpu': f'{extra_models_cpu}',\n",
    " '--esrgan-cpu': f'{esrgan_cpu}',\n",
    " '--gfpgan-cpu': f'{gfpgan_cpu}'}\n",
    "\n",
    "only_true_vars = {k for (k,v) in run_string_with_variables.items() if v == 'True'}\n",
    "vars = \" \".join(only_true_vars)\n"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "E1kYjBY9Rmi-"
   },
   "source": [
    "## 3 - Launch WebUI for stable diffusion"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "cellView": "form",
    "id": "BTH_drY9KZ4k"
   },
   "outputs": [],
   "source": [
    "#@markdown ** keep in mind that this script is set to run for ever, google will disconnect you after 90 minutes on free tiers\n",
    "\n",
    "#@markdown # Important - click the public URL to launch WebUI in another tab\n",
    "#@markdown ![](https://user-images.githubusercontent.com/463317/187105407-dd9b0f4e-c8da-49d3-8c78-1767f5c9aa83.jpg)\n",
    "\n",
    "#fix adding share_password to the launch params, and also changin {vars} to $vars as it was causing webui.py to fail.\n",
    "\n",
    "%cd /content/stable-diffusion\n",
    "\n",
    "if share_password == \"\":\n",
    "  !python scripts/webui.py \\\n",
    "  --ckpt '{models_path}/sd-v1-4.ckpt' \\\n",
    "  --outdir '{output_path}' \\\n",
    "  --share $vars\n",
    "else:\n",
    "  !python scripts/webui.py \\\n",
    "  --ckpt '{models_path}/sd-v1-4.ckpt' \\\n",
    "  --outdir '{output_path}' \\\n",
    "  --share-password '{share_password}' \\\n",
    "  --share $vars\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "id": "iZ-X7HxeRNRQ"
   },
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "accelerator": "GPU",
  "colab": {
   "collapsed_sections": [
    "4eodrBxeCgOW",
    "RXasLtrYC9uj",
    "MVJzvQGdDn6C",
    "vhZQqFGxEHcA",
    "Wt_tzv_eJ9Qz",
    "ip0o1IqmQNON",
    "E1kYjBY9Rmi-"
   ],
   "machine_shape": "hm",
   "name": "Stable Diffusion WebUi - Altryne",
   "provenance": []
  },
  "gpuClass": "standard",
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "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.9.13"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}