fast_stable_diffusion_AUTOMATIC1111.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# **Colab From https://github.com/TheLastBen/fast-stable-diffusion, if you face any issues, feel free to discuss them.** [Support](https://ko-fi.com/thelastben)\n",
    "\n",
    "\n",
    "\n",
    "\n",
    "\n",
    "\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {
    "id": "BQ_jNgfxb6H6"
   },
   "outputs": [],
   "source": [
    "#@title Colab/Local settings\n",
    "#@markdown # Do you want to use google drive?\n",
    "import os\n",
    "\n",
    "use_gdrive = False #@param {type:\"boolean\"}\n",
    "#@markdown Note: Google Drive only works on Google Colab hosted runtimes\n",
    "if use_gdrive:\n",
    "  try:\n",
    "    from google.colab import drive\n",
    "    drive.mount('/content/gdrive')\n",
    "    working_dir = '/content/gdrive/MyDrive'\n",
    "  except:\n",
    "    print('It appears that this is not running on Google Colab, disabling gdrive')\n",
    "    working_dir = os.getcwd()\n",
    "\n",
    "else:\n",
    "  working_dir = os.getcwd()\n",
    "\n",
    "# Get site-packages (can't be named site-packages)\n",
    "from subprocess import getoutput\n",
    "site_packages = getoutput('pip show torch')\n",
    "site_packages = site_packages[site_packages.find('Location:')+10:]\n",
    "site_packages = site_packages[:site_packages.find('\\n')]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/"
    },
    "id": "CFWtw-6EPrKi",
    "outputId": "86d6328e-a4e7-4b46-b9cf-57420c2574e4"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "/home/ben/big-stuff/ai-and-notebooks/stable-diffusion/universal-fast-stable-diffusion/sd\n"
     ]
    }
   ],
   "source": [
    "#@markdown # Installing AUTOMATIC1111 repo\n",
    "%mkdir sd\n",
    "%cd sd\n",
    "from IPython.utils import capture\n",
    "from IPython.display import clear_output\n",
    "\n",
    "with capture.capture_output() as cap:\n",
    "  fgitclone = \"git clone --depth 1\" \n",
    "  %cd $working_dir/\n",
    "  %mkdir  sd\n",
    "  %cd sd\n",
    "  !$fgitclone https://github.com/Stability-AI/stablediffusion\n",
    "  !$fgitclone https://github.com/AUTOMATIC1111/stable-diffusion-webui\n",
    "  %cd $working_dir/sd/stable-diffusion-webui/\n",
    "  !mkdir -p cache/{huggingface,torch}\n",
    "  %cd $working_dir\n",
    "  !ln -s $working_dir/sd/stable-diffusion-webui/cache/huggingface ../root/.cache/\n",
    "  !ln -s $working_dir/sd/stable-diffusion-webui/cache/torch ../root/.cache/\n",
    "  %pip install -q -U gdown\n",
    "\n",
    "Update_repo = False #@param {type:\"boolean\"}\n",
    "if Update_repo:  \n",
    "  !rm -f $working_dir/sd/stable-diffusion-webui/webui.sh\n",
    "  !rm -f $working_dir/sd/stable-diffusion-webui/modules/paths.py\n",
    "  !rm -f $working_dir/sd/stable-diffusion-webui/webui.py \n",
    "  !rm -f $working_dir/sd/stable-diffusion-webui/modules/ui.py\n",
    "  !rm -f $working_dir/sd/stable-diffusion-webui/style.css\n",
    "  %cd $working_dir/sd/stable-diffusion-webui/\n",
    "  clear_output()\n",
    "  print('\u001b[1;32m')\n",
    "  !git pull"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 25,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/"
    },
    "id": "p4wj_txjP3TC",
    "outputId": "ac0d5386-9428-4711-f90f-9102c485fb34"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Reinitialized existing Git repository in /home/ben/big-stuff/ai-and-notebooks/stable-diffusion/universal-fast-stable-diffusion/.git/\n",
      "warning: current user is not root/admin, system install is likely to fail.\n",
      "Git LFS initialized.\n",
      "Cloning into 'stable-diffusion-v1-5'...\n",
      "remote: Enumerating objects: 26, done.\u001b[K\n",
      "remote: Counting objects: 100% (26/26), done.\u001b[K\n",
      "remote: Compressing objects: 100% (25/25), done.\u001b[K\n",
      "remote: Total 26 (delta 1), reused 0 (delta 0), pack-reused 0\u001b[K\n",
      "Unpacking objects: 100% (26/26), 520.92 KiB | 451.00 KiB/s, done.\n",
      "Filtering content: 100% (4/4), 2.55 GiB | 25.05 MiB/s, done.\n",
      "\u001b[1;31mMake sure you accept the terms at https://huggingface.co/runwayml/stable-diffusion-v1-5\n"
     ]
    }
   ],
   "source": [
    "import os\n",
    "from IPython.display import clear_output\n",
    "import time\n",
    "\n",
    "#@markdown # Model Download/Load\n",
    "Model_Version = \"1.5\" #@param [\"V2\", \"1.5\"]\n",
    "\n",
    "RunwayML_Inpainting_Model = False #@param {type:\"boolean\"}\n",
    "\n",
    "token = \"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\" #@param {type:\"string\"}\n",
    "\n",
    "Redownload_the_original_model = True #@param {type:\"boolean\"}\n",
    "\n",
    "if Redownload_the_original_model:\n",
    "  if os.path.exists(f'{working_dir}/mainmodel.ckpt'):\n",
    "    !rm -f $working_dir/mainmodel.ckpt\n",
    "  %cd $working_dir/sd/stable-diffusion-webui/models/Stable-diffusion  \n",
    "  !wget -q -O model.ckpt https://raw.githubusercontent.com/TheLastBen/fast-stable-diffusion/main/precompiled/attention.py\n",
    "  !mv $working_dir/sd/stable-diffusion-webui/models/Stable-diffusion/model.ckpt $working_dir/sd/stable-diffusion-webui/models/Stable-diffusion/trashfile.f  \n",
    "  time.sleep(2)\n",
    "  !rm -f $working_dir/sd/stable-diffusion-webui/models/Stable-diffusion/trashfile.f\n",
    "  time.sleep(2)\n",
    "  clear_output()\n",
    "\n",
    "#@markdown Or\n",
    "Path_to_CKPT = \"\" #@param {type:\"string\"}\n",
    "#@markdown - Insert the full path of your trained model (eg: /content/gdrive/MyDrive/zarathustra.ckpt) or to a folder containing multiple models.\n",
    "\n",
    "#@markdown Or\n",
    "Link_CKPT = \"\" #@param {type:\"string\"}\n",
    "#@markdown - A direct link to a CKPT or a shared gdrive link.\n",
    "\n",
    "def newmdl(token):\n",
    "\n",
    "    if not os.path.exists(f'{working_dir}/sd/stable-diffusion-webui/models/Stable-diffusion/model.ckpt'):\n",
    "      if token==\"\":\n",
    "        token=input(\"Insert your huggingface token :\")\n",
    "      %cd $working_dir\n",
    "      clear_output()\n",
    "      !git init\n",
    "      !git lfs install --system --skip-repo\n",
    "      !$fgitclone --branch fp16 \"https://USER:{token}@huggingface.co/runwayml/stable-diffusion-v1-5\"\n",
    "      if os.path.exists(f'{working_dir}/stable-diffusion-v1-5'):\n",
    "        !$fgitclone \"https://USER:{token}@huggingface.co/stabilityai/sd-vae-ft-mse\"\n",
    "        !rm -rf $working_dir/stable-diffusion-v1-5/vae\n",
    "        !mv $working_dir/sd-vae-ft-mse $working_dir/stable-diffusion-v1-5/vae        \n",
    "        !wget -O convertosd.py https://github.com/TheLastBen/fast-stable-diffusion/raw/main/Dreambooth/convertosd.py\n",
    "        !sed -i '201s@.*@    model_path = \"{working_dir}/stable-diffusion-v1-5\"@' $working_dir/convertosd.py\n",
    "        !sed -i '202s@.*@    checkpoint_path= \"{working_dir}/sd/stable-diffusion-webui/models/Stable-diffusion/model.ckpt\"@' $working_dir/convertosd.py\n",
    "        clear_output()       \n",
    "        !python $working_dir/convertosd.py \n",
    "        if os.path.exists(f'{working_dir}/sd/stable-diffusion-webui/models/Stable-diffusion/model.ckpt'):\n",
    "          model=f'{working_dir}/sd/stable-diffusion-webui/models/Stable-diffusion/model.ckpt'\n",
    "          clear_output()\n",
    "          print('\u001b[1;32mDONE !')\n",
    "        else:\n",
    "          print('\u001b[1;31mSomething went wrong, try again')\n",
    "      else:\n",
    "        print('\u001b[1;31mMake sure you accept the terms at https://huggingface.co/runwayml/stable-diffusion-v1-5')\n",
    "\n",
    "    elif os.path.exists(f'{working_dir}/sd/stable-diffusion-webui/models/Stable-diffusion/model.ckpt'):\n",
    "      model=f'{working_dir}/sd/stable-diffusion-webui/models/Stable-diffusion/model.ckpt'\n",
    "      clear_output()\n",
    "      print('\u001b[1;32mModel already exists, check the box \"Redownload_the_original_model\" to redownload/download the V1.5')\n",
    "\n",
    "    if os.path.exists(f'{working_dir}/.git'):\n",
    "      !rm -rf $working_dir/.git\n",
    "\n",
    "    if os.path.exists(f'{working_dir}/stable-diffusion-v1-5'):\n",
    "      !rm -rf $working_dir/stable-diffusion-v1-5      \n",
    "\n",
    "\n",
    "def V2(token):\n",
    "    if not os.path.exists(f'{working_dir}/sd/stable-diffusion-webui/models/Stable-diffusion/model.ckpt'):\n",
    "      %cd $working_dir/\n",
    "      clear_output()\n",
    "      !mkdir \"{working_dir}/stable-diffusion-V2\"\n",
    "      %cd \"{working_dir}/stable-diffusion-V2\"\n",
    "      !git init\n",
    "      !git lfs install --system --skip-repo\n",
    "      !git remote add -f origin  \"https://USER:{token}@huggingface.co/stabilityai/stable-diffusion-2\"\n",
    "      !git config core.sparsecheckout true\n",
    "      !echo -e \"scheduler\\ntext_encoder\\ntokenizer\\nunet\\nvae\\nmodel_index.json\" > .git/info/sparse-checkout\n",
    "      !git pull origin fp16\n",
    "      %cd $working_dir\n",
    "      !wget -O convertosd.py https://github.com/TheLastBen/fast-stable-diffusion/raw/main/Dreambooth/convertosd.py\n",
    "      !sed -i '201s@.*@    model_path = \"{working_dir}/stable-diffusion-V2\"@' $working_dir/convertosd.py\n",
    "      !sed -i '202s@.*@    checkpoint_path= \"{working_dir}/gdrive/MyDrive/sd/stable-diffusion-webui/models/Stable-diffusion/model.ckpt\"@' $working_dir/convertosd.py\n",
    "      clear_output()       \n",
    "      !python $working_dir/convertosd.py\n",
    "      if os.path.exists(f'{working_dir}/sd/stable-diffusion-webui/models/Stable-diffusion/model.ckpt'):\n",
    "        model=f'{working_dir}/sd/stable-diffusion-webui/models/Stable-diffusion/model.ckpt'\n",
    "        clear_output()\n",
    "        print('\u001b[1;32mDONE !')\n",
    "      else:\n",
    "        print('\u001b[1;31mSomething went wrong, try again')\n",
    "\n",
    "\n",
    "    elif os.path.exists(f'{working_dir}/sd/stable-diffusion-webui/models/Stable-diffusion/model.ckpt'):\n",
    "      model=f'{working_dir}/sd/stable-diffusion-webui/models/Stable-diffusion/model.ckpt'\n",
    "      clear_output()\n",
    "      print('\u001b[1;32mModel already exists, check the box \"Redownload_the_original_model\" to redownload/download the V2.')\n",
    "\n",
    "    if os.path.exists(f'{working_dir}/.git'):\n",
    "      !rm -rf $working_dir/.git\n",
    "\n",
    "    if os.path.exists(f'{working_dir}/stable-diffusion-V2'):\n",
    "      !rm -rf $working_dir/stable-diffusion-V2\n",
    "\n",
    "def inpmdl(token):\n",
    "\n",
    "    if not os.path.exists(f'{working_dir}/sd/stable-diffusion-webui/models/Stable-diffusion/sd-v1-5-inpainting.ckpt'):\n",
    "      if token==\"\":\n",
    "        token=input(\"Insert your huggingface token :\")\n",
    "      %cd $working_dir/\n",
    "      clear_output()\n",
    "      !git init\n",
    "      !git lfs install --system --skip-repoInference API has been turned off for this model.\n",
    "      !$fgitclone --branch fp16 \"https://USER:{token}@huggingface.co/runwayml/stable-diffusion-inpainting\"\n",
    "      if os.path.exists(f'{working_dir}/stable-diffusion-inpainting'):\n",
    "        !$fgitclone \"https://USER:{token}@huggingface.co/stabilityai/sd-vae-ft-mse\"\n",
    "        !rm -rf $working_dir/stable-diffusion-inpainting/vae\n",
    "        !mv $working_dir/sd-vae-ft-mse $working_dir/stable-diffusion-inpainting/vae        \n",
    "        !wget -O convertosd.py https://github.com/TheLastBen/fast-stable-diffusion/raw/main/Dreambooth/convertosd.py\n",
    "        !sed -i '201s@.*@    model_path = \"{working_dir}/stable-diffusion-inpainting\"@' $working_dir/convertosd.py\n",
    "        !sed -i '202s@.*@    checkpoint_path= \"{working_dir}/sd/stable-diffusion-webui/models/Stable-diffusion/sd-v1-5-inpainting.ckpt\"@' $working_dir/convertosd.py\n",
    "        clear_output()       \n",
    "        !python $working_dir/convertosd.py \n",
    "        if os.path.exists(f'{working_dir}/sd/stable-diffusion-webui/models/Stable-diffusion/sd-v1-5-inpainting.ckpt'):\n",
    "          clear_output()\n",
    "          print('\u001b[1;32mDONE !')\n",
    "        else:\n",
    "          print('\u001b[1;31mSomething went wrong, try again')\n",
    "      else:\n",
    "        print('\u001b[1;31mMake sure you have accepted the terms at https://huggingface.co/runwayml/stable-diffusion-inpainting')\n",
    "\n",
    "    elif os.path.exists(f'{working_dir}/sd/stable-diffusion-webui/models/Stable-diffusion/sd-v1-5-inpainting.ckpt'):\n",
    "      clear_output()\n",
    "      print('\u001b[1;32mModel already exists.')\n",
    "\n",
    "    if os.path.exists(f'{working_dir}/.git'):\n",
    "      !rm -rf $working_dir/.git\n",
    "\n",
    "    if os.path.exists(f'{working_dir}/stable-diffusion-inpainting'):\n",
    "      !rm -rf $working_dir/stable-diffusion-inpainting   \n",
    "\n",
    "if (Path_to_CKPT !=''):\n",
    "  if os.path.exists(str(Path_to_CKPT)):\n",
    "    print('\u001b[1;32mUsing the trained model.')\n",
    "  else:\n",
    "      while not os.path.exists(str(Path_to_CKPT)):\n",
    "        print('\u001b[1;31mWrong path, use the colab file explorer to copy the path : ')\n",
    "        Path_to_CKPT=input()\n",
    "      if os.path.exists(str(Path_to_CKPT)):\n",
    "        print('\u001b[1;32mUsing the trained model.')\n",
    "\n",
    "  model=Path_to_CKPT\n",
    "\n",
    "elif Link_CKPT != \"\":\n",
    "  if os.path.exists(f'{working_dir}/mainmodel.ckpt'):\n",
    "    !rm -f $working_dir/mainmodel.ckpt\n",
    "  %cd $working_dir/sd/stable-diffusion-webui/models/Stable-diffusion\n",
    "  !wget -q -O model.ckpt https://raw.githubusercontent.com/TheLastBen/fast-stable-diffusion/main/precompiled/attention.py\n",
    "  !mv $working_dir/sd/stable-diffusion-webui/models/Stable-diffusion/model.ckpt $working_dir/sd/stable-diffusion-webui/models/Stable-diffusion/trashfile.f  \n",
    "  time.sleep(2)\n",
    "  !rm -f $working_dir/sd/stable-diffusion-webui/models/Stable-diffusion/trashfile.f\n",
    "  time.sleep(2)\n",
    "  clear_output()\n",
    "  !gdown --fuzzy $Link_CKPT -O model.ckpt\n",
    "  if os.path.exists(f'{working_dir}/sd/stable-diffusion-webui/models/Stable-diffusion/model.ckpt'):\n",
    "    if os.path.getsize(f\"{working_dir}/sd/stable-diffusion-webui/models/Stable-diffusion/model.ckpt\") > 1810671599:\n",
    "      model=f'{working_dir}/sd/stable-diffusion-webui/models/Stable-diffusion/model.ckpt'\n",
    "      clear_output()\n",
    "      print('\u001b[1;32mModel downloaded, using the trained model.')\n",
    "    else:\n",
    "      print('\u001b[1;31mWrong link, check that the link is valid')\n",
    "  else:\n",
    "    print('\u001b[1;31mWrong link, check that the link is valid')\n",
    "\n",
    "\n",
    "elif Model_Version==\"1.5\":\n",
    "  newmdl(token)\n",
    "\n",
    "elif Model_Version==\"V2\":\n",
    "  V2(token)\n",
    "\n",
    "if RunwayML_Inpainting_Model:\n",
    "  inpmdl(token)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {
    "id": "ZGV_5H4xrOSp"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\u001b[1;32mDone!\n"
     ]
    }
   ],
   "source": [
    "#@markdown # Requirements\n",
    "\n",
    "import os\n",
    "import time\n",
    "from IPython.utils import capture\n",
    "from IPython.display import clear_output\n",
    "from IPython.display import HTML\n",
    "from subprocess import getoutput\n",
    "from re import search\n",
    "\n",
    "with capture.capture_output() as cap:\n",
    "  if not os.path.exists(f'{working_dir}/sd/stablediffusion/src/k-diffusion/k_diffusion'):\n",
    "    !mkdir $working_dir/sd/stablediffusion/src\n",
    "    %cd $working_dir/sd/stablediffusion/src\n",
    "    !$fgitclone https://github.com/CompVis/taming-transformers\n",
    "    !$fgitclone https://github.com/openai/CLIP\n",
    "    !$fgitclone https://github.com/salesforce/BLIP\n",
    "    !$fgitclone https://github.com/sczhou/CodeFormer\n",
    "    !$fgitclone https://github.com/crowsonkb/k-diffusion\n",
    "    !mv $working_dir/sd/stablediffusion/src/CLIP $working_dir/sd/stablediffusion/src/clip\n",
    "    !mv  $working_dir/sd/stablediffusion/src/BLIP $working_dir/sd/stablediffusion/src/blip    \n",
    "    !mv  $working_dir/sd/stablediffusion/src/CodeFormer $working_dir/sd/stablediffusion/src/codeformer        \n",
    "    !cp -r $working_dir/sd/stablediffusion/src/k-diffusion/k_diffusion $working_dir/sd/stable-diffusion-webui/\n",
    "\n",
    "  %cd $working_dir/\n",
    "  !wget https://github.com/TheLastBen/fast-stable-diffusion/raw/main/Dependencies/Dependencies_AUT.1\n",
    "  !wget https://github.com/TheLastBen/fast-stable-diffusion/raw/main/Dependencies/Dependencies_AUT.2\n",
    "  !mv Dependencies_AUT.1 Dependencies_AUT.7z.001\n",
    "  !mv Dependencies_AUT.2 Dependencies_AUT.7z.002\n",
    "  !7z x Dependencies_AUT.7z.001\n",
    "  time.sleep(3)\n",
    "  !cp -r $working_dir/usr/local/lib/python3.7/dist-packages $site_packages\n",
    "  !rm -rf $working_dir/usr\n",
    "  !rm -f Dependencies_AUT.7z.001\n",
    "  !rm -f Dependencies_AUT.7z.002\n",
    "  %cd $working_dir/sd/stable-diffusion-webui/modules\n",
    "  !wget -O paths.py https://raw.githubusercontent.com/askiiart/fast-stable-diffusion/main/AUTOMATIC1111_files/paths.py\n",
    "  %pip install -q gradio==3.9\n",
    "  %pip install -q open_clip_torch\n",
    "  %pip install -q torchsde\n",
    "  %pip install -q scikit-image --upgrade\n",
    "  %pip install -q tensorflow tensorflow-io\n",
    "  %pip install -q git+https://github.com/KichangKim/DeepDanbooru.git@edf73df4cdaeea2cf00e9ac08bd8a9026b7a7b26#egg=deepdanbooru\n",
    "  %pip install -q inflection\n",
    "\n",
    "  if os.path.exists(f'{working_dir}/sd/stable-diffusion-webui/extensions/deforum-for-automatic1111-webui/scripts/deforum_helpers/src/infer.py'):\n",
    "    %cd $working_dir/sd/stable-diffusion-webui/extensions/deforum-for-automatic1111-webui/scripts/deforum_helpers\n",
    "    !wget -O animation.py https://raw.githubusercontent.com/TheLastBen/fast-stable-diffusion/main/AUTOMATIC1111_files/deforum/animation.py\n",
    "    !wget -O depth.py https://raw.githubusercontent.com/TheLastBen/fast-stable-diffusion/main/AUTOMATIC1111_files/deforum/depth.py    \n",
    "    %cd $working_dir/sd/stable-diffusion-webui/extensions/deforum-for-automatic1111-webui/scripts/deforum_helpers/src    \n",
    "    !wget -O infer.py https://raw.githubusercontent.com/TheLastBen/fast-stable-diffusion/main/AUTOMATIC1111_files/deforum/infer.py\n",
    "    %cd $working_dir\n",
    "  \n",
    "def xformers() -> None:\n",
    "  smi_out = getoutput('nvidia-smi')\n",
    "  supported = search('(T4|P100|V100|A100|K80)', smi_out)\n",
    "\n",
    "  if not supported:\n",
    "    while True:\n",
    "      print(\"\\x1b[1;31mThe current GPU is not supported, try starting a new session.\\x1b[0m\")\n",
    "      time.sleep(5)\n",
    "  else:\n",
    "    supported = supported.group(0)\n",
    "\n",
    "  %pip install -q https://github.com/TheLastBen/fast-stable-diffusion/raw/main/precompiled/{supported}/xformers-0.0.13.dev0-py3-none-any.whl\n",
    "\n",
    "  clear_output()\n",
    "\n",
    "def cleanup():\n",
    "    base = f'{working_dir}/sd/stable-diffusion'\n",
    "    dirs = [base, f'{base}/src/taming-transformers', f'{base}/src/clip',\n",
    "            f'{base}/src/GFPGAN', f'{base}/src/blip', f'{base}/src/codeformer',\n",
    "            f'{base}/src/realesrgan', f'{base}/src/k-diffusion', f'{base}/src/ldm']\n",
    "    for d in dirs:\n",
    "        !rm -rf {d + '/.git'}\n",
    "    print('\u001b[1;32mDone!')\n",
    "\n",
    "xformers()\n",
    "cleanup()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 23,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/"
    },
    "id": "PjzwxTkPSPHf",
    "outputId": "70b1afcc-8e28-4c82-eb98-d1bdae0a0dcb"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "grep: /home/ben/big-stuff/ai-and-notebooks/stable-diffusion/universal-fast-stable-diffusion/srv.txt: No such file or directory\n",
      "sed: can't read /home/ben/.local/lib/python3.10/site-packages/gradio/blocks.py: No such file or directory\n",
      "sed: can't read /home/ben/.local/lib/python3.10/site-packages/gradio/blocks.py: No such file or directory\n",
      "sed: can't read /home/ben/.local/lib/python3.10/site-packages/gradio/blocks.py: No such file or directory\n",
      "sed: can't read /home/ben/.local/lib/python3.10/site-packages/gradio/strings.py: No such file or directory\n",
      "rm: cannot remove '/home/ben/big-stuff/ai-and-notebooks/stable-diffusion/universal-fast-stable-diffusion/srv.txt': No such file or directory\n",
      "No checkpoints found. When searching for checkpoints, looked at:\n",
      " - file /home/ben/big-stuff/ai-and-notebooks/stable-diffusion/universal-fast-stable-diffusion/sd/stable-diffusion-webui/model.ckpt\n",
      " - directory /home/ben/big-stuff/ai-and-notebooks/stable-diffusion/universal-fast-stable-diffusion/sd/stable-diffusion-webui/models/Stable-diffusion\n",
      "Can't run without a checkpoint. Find and place a .ckpt file into any of those locations. The program will exit.\n"
     ]
    }
   ],
   "source": [
    "#@markdown # Start stable-diffusion\n",
    "from IPython.utils import capture\n",
    "from subprocess import getoutput\n",
    "import time\n",
    "import sys\n",
    "import fileinput\n",
    "\n",
    "V2 = False #@param {type:\"boolean\"}\n",
    "#@markdown  - Check if the model is v2\n",
    "\n",
    "Use_Gradio_Server = False #@param {type:\"boolean\"}\n",
    "#@markdown  - Only if you have trouble connecting to the local server\n",
    "\n",
    "Hypernetwork_Compatibility = False #@param {type:\"boolean\"}\n",
    "\n",
    "Enable_API = False #@param {type:\"boolean\"}\n",
    "#@markdown  - You can find the docs in the [wiki](https://github.com/AUTOMATIC1111/stable-diffusion-webui/wiki/API)\n",
    "\n",
    "with capture.capture_output() as cap:\n",
    "  smi_out = getoutput('nvidia-smi')\n",
    "  if Hypernetwork_Compatibility:\n",
    "    %pip uninstall -y xformers\n",
    "  else:\n",
    "    if not os.path.exists(f'{site_packages}/xformers'):\n",
    "      xformers()   \n",
    "  \n",
    "with capture.capture_output() as cap: \n",
    "  if not os.path.exists('/tools/node/bin/lt'):\n",
    "    !npm install -g localtunnel\n",
    "\n",
    "with capture.capture_output() as cap: \n",
    "  %cd $working_dir/sd/stable-diffusion-webui/\n",
    "  time.sleep(1)\n",
    "  !wget -O webui.py https://raw.githubusercontent.com/AUTOMATIC1111/stable-diffusion-webui/master/webui.py\n",
    "  !sed -i 's@gpu_call).*@gpu_call) \\n        demo.queue(concurrency_count=111500)@' $working_dir/sd/stable-diffusion-webui/webui.py\n",
    "  %cd $working_dir/sd/stable-diffusion-webui/modules/\n",
    "  !wget -O ui.py https://raw.githubusercontent.com/AUTOMATIC1111/stable-diffusion-webui/master/modules/ui.py\n",
    "  !sed -i 's@css = \"\".*@with open(os.path.join(script_path, \"style.css\"), \"r\", encoding=\"utf8\") as file:\\n        css = file.read()@' $working_dir/sd/stable-diffusion-webui/modules/ui.py\n",
    "  %cd $working_dir/sd/stable-diffusion-webui\n",
    "  !wget -O style.css https://raw.githubusercontent.com/AUTOMATIC1111/stable-diffusion-webui/master/style.css\n",
    "  !sed -i 's@min-height: 4.*@min-height: 5.5em;@g' $working_dir/sd/stable-diffusion-webui/style.css  \n",
    "  %cd $working_dir\n",
    "\n",
    "\n",
    "share=''\n",
    "if Use_Gradio_Server:\n",
    "  share='--share'\n",
    "  for line in fileinput.input(f'{site_packages}/gradio/blocks.py', inplace=True):\n",
    "    if line.strip().startswith('self.server_name ='):\n",
    "        line = '            self.server_name = server_name\\n'\n",
    "    if line.strip().startswith('self.server_port ='):\n",
    "        line = '            self.server_port = server_port\\n'\n",
    "    sys.stdout.write(line)\n",
    "  clear_output()\n",
    "\n",
    "with capture.capture_output() as cap: \n",
    "  %cd $working_dir/sd/stablediffusion\n",
    "\n",
    "api = '--api' if Enable_API else ''\n",
    "\n",
    "if V2:\n",
    "  configf=f\"--config {working_dir}/sd/stablediffusion/configs/stable-diffusion/v2-inference-v.yaml\"\n",
    "else:\n",
    "  configf=\"\"\n",
    "\n",
    "try:\n",
    "  model\n",
    "  if os.path.isfile(model):\n",
    "    !python $working_dir/sd/stable-diffusion-webui/webui.py $share $api --disable-safe-unpickle --no-half-vae  --ckpt \"$model\" $configf\n",
    "  else:\n",
    "    !python $working_dir/sd/stable-diffusion-webui/webui.py $share $api --disable-safe-unpickle --no-half-vae  --ckpt-dir \"$model\" $configf\n",
    "except:\n",
    "   !python $working_dir/sd/stable-diffusion-webui/webui.py $share $api --disable-safe-unpickle --no-half-vae $configf"
   ]
  }
 ],
 "metadata": {
  "accelerator": "GPU",
  "colab": {
   "collapsed_sections": [],
   "provenance": []
  },
  "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.10.6"
  },
  "vscode": {
   "interpreter": {
    "hash": "d4c34f896473ed2a16a8fbd446cd50d87ccecb3be42ee3d3759843c10ee7267b"
   }
  }
 },
 "nbformat": 4,
 "nbformat_minor": 1
}