stablediffusion_runpod_adapted_webgui.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "a9857407-8bc3-494f-8d61-814bda753f29",
   "metadata": {},
   "source": [
    "# Stable Diffusion WebUI\n",
    "*Optimized for RunPod*"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "26f0fb1d-4b32-4c50-b4bc-ce6cdbc8b184",
   "metadata": {},
   "source": [
    "This notebook can be run end-to-end to get a working WebUI for Stable Diffusion with support for face correction (GFPGAN) and superresolution upscaling (ESRGAN).\n",
    "\n",
    "If running on RunPod, assumes you're starting with the RunPod Pytorch container. That said, this notebook does not make use of the built-in preinstalled pytorch as version/dependency incompatibilities between the preinstalled packages and the ones needed by the pipeline make it much easier to start fresh.\n",
    "\n",
    "This notebook is designed to create a persistent conda environment and model files in the /workspace/ directory, which persists across stopping and starting pods. It can easily be modified to use any similar persistent storage in other services (e.g. a google drive mount in colab).\n",
    "\n",
    "Ensure you're on a GPU with the appropriate VRAM for what you're trying to generate. GPUs with 16GB or less VRAM will probably only be able to generate a single 512x512 image at a time. Larger GPUs like the A100 are recommended but definitely more expensive."
   ]
  },
  {
   "cell_type": "markdown",
   "id": "51f8b693",
   "metadata": {},
   "source": [
    "***"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "7135bb5e",
   "metadata": {},
   "source": [
    "## 1. Set Environment variables and configure your web-app\n",
    "By default, this notebook assumes you're running in RunPod and will create all resources in the `/workspace/` directory which persists across runs. Feel free to change this to your preferred path. Using a persistent location is recommended (e.g. a Google Drive mount in Colab or a non-ephemeral EBS volume in AWS) so that you don't have to reinstall and reconfigure everything every time you restart your environment.\n",
    "\n",
    "### **The cells in this section should be run every time you start the notebook**"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "4b782c5e",
   "metadata": {},
   "source": [
    "### Environment Variables"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "c5e51e81",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Set your working directory and data locations here (usually whatever persistent storage mount is available to your runtime)\n",
    "#For RunPod it will be /workspace/\n",
    "wd = \"/workspace\"\n",
    "models_path = wd +\"/models\"\n",
    "output_path = wd +\"/output\"\n",
    "\n",
    "# Set whether or not you want GFPGAN and ESRGAN here\n",
    "add_GFP = True # Determines whether or not GFPGAN (face correction) will be installed\n",
    "add_ESR = True # Determines whether or not ESRGAN (resolution upscaling) will be installed\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "6c675dbc",
   "metadata": {},
   "outputs": [],
   "source": [
    "\n",
    "conda_env = wd + \"/sd_env\"\n",
    "pip_path = conda_env + \"/bin/pip\"\n",
    "python_path = conda_env + \"/bin/python\"\n",
    "%cd {wd}"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "fad2d441",
   "metadata": {},
   "source": [
    "### Web UI and Stable Diffusion Options"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "202d37e7",
   "metadata": {},
   "source": [
    "Set your launch preferences here. Intended for advanced users. The defaults should work for almost everyone."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "876502ca",
   "metadata": {},
   "outputs": [],
   "source": [
    "share_password=\"\" # Add a password to your webui\n",
    "defaults=\"configs/webui/webui.yaml\" # path to configuration file providing UI defaults, uses same format as cli parameter  \n",
    "\n",
    "save_metadata = False # Whether to embed the generation parameters in the sample images\n",
    "\n",
    "skip_grid = False # Do not save a grid, only individual samples. Helpful when evaluating lots of samples\n",
    "skip_save = False # Do not save individual samples as files. For speed measurements\n",
    "optimized = False # 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 # Alternative optimization mode that does not save as much VRAM but runs siginificantly faster\n",
    "no_verify_input = False # Do not verify input to check if it's too long\n",
    "no_half = False # Do not switch the model to 16-bit floats\n",
    "no_progressbar_hiding = True # Do not hide progressbar in gradio UI\n",
    "extra_models_cpu = False # Run extra models (GFGPAN/ESRGAN) on cpu\n",
    "esrgan_cpu = True # run ESRGAN on cpu\n",
    "gfpgan_cpu = True # run GFPGAN on cpu"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "646af61b",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Set up the parameters that will be passed to the web UI later\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)"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "940a11eb",
   "metadata": {},
   "source": [
    "***"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "2a7a8708",
   "metadata": {},
   "source": [
    "## 2. Install and set up all dependencies\n",
    "This section takes care of configuration of your environment. \n",
    "### **If you've already done this on your persistent storage in a previous run, you can skip this whole section and move to section 3**"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "081f82fe-87f7-4a7e-948b-f42ca528e451",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "GPU 0: NVIDIA A100 80GB PCIe (UUID: GPU-d9c7c489-f1a8-03ff-52c7-a405cb020992)\n"
     ]
    }
   ],
   "source": [
    "# Check CUDA stats to make sure we have the correct GPU\n",
    "!nvidia-smi -L"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "b4ec785e-09d6-4b65-9597-0ba5dc6ad29f",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\n",
      "\n",
      "WARNING: The --force flag will be removed in a future conda release.\n",
      "         See 'conda update --help' for details about the --force-reinstall\n",
      "         and --clobber flags.\n",
      "\n",
      "\n",
      "Collecting package metadata (current_repodata.json): done\n",
      "Solving environment: done\n",
      "\n",
      "## Package Plan ##\n",
      "\n",
      "  environment location: /opt/conda\n",
      "\n",
      "  added / updated specs:\n",
      "    - conda\n",
      "\n",
      "\n",
      "The following packages will be downloaded:\n",
      "\n",
      "    package                    |            build\n",
      "    ---------------------------|-----------------\n",
      "    _openmp_mutex-5.1          |            1_gnu          21 KB\n",
      "    ca-certificates-2022.07.19 |       h06a4308_0         124 KB\n",
      "    cffi-1.15.1                |   py37h74dc2b5_0         227 KB\n",
      "    conda-4.14.0               |   py37h06a4308_0         909 KB\n",
      "    cytoolz-0.11.0             |   py37h7b6447c_0         328 KB\n",
      "    ffmpeg-4.2.2               |       h20bf706_0        59.6 MB\n",
      "    ld_impl_linux-64-2.38      |       h1181459_1         654 KB\n",
      "    lerc-3.0                   |       h295c915_0         196 KB\n",
      "    libdeflate-1.8             |       h7f8727e_5          51 KB\n",
      "    libgcc-ng-11.2.0           |       h1234567_1         5.3 MB\n",
      "    libgomp-11.2.0             |       h1234567_1         474 KB\n",
      "    libopus-1.3.1              |       h7b6447c_0         491 KB\n",
      "    libstdcxx-ng-11.2.0        |       h1234567_1         4.7 MB\n",
      "    libtiff-4.4.0              |       hecacb30_0         471 KB\n",
      "    libvpx-1.7.0               |       h439df22_0         1.2 MB\n",
      "    ncurses-6.3                |       h5eee18b_3         781 KB\n",
      "    openssl-1.1.1q             |       h7f8727e_0         2.5 MB\n",
      "    pillow-9.2.0               |   py37hace64e9_1         664 KB\n",
      "    pip-22.1.2                 |   py37h06a4308_0         2.4 MB\n",
      "    psutil-5.9.0               |   py37h5eee18b_0         328 KB\n",
      "    requests-2.28.1            |   py37h06a4308_0          92 KB\n",
      "    ripgrep-13.0.0             |       hbdeaff8_0         1.1 MB\n",
      "    setuptools-63.4.1          |   py37h06a4308_0         1.1 MB\n",
      "    sqlite-3.39.2              |       h5082296_0         1.1 MB\n",
      "    tk-8.6.12                  |       h1ccaba5_0         3.0 MB\n",
      "    toolz-0.11.2               |     pyhd3eb1b0_0          49 KB\n",
      "    tqdm-4.64.0                |   py37h06a4308_0         126 KB\n",
      "    typing_extensions-4.3.0    |   py37h06a4308_0          42 KB\n",
      "    urllib3-1.26.11            |   py37h06a4308_0         181 KB\n",
      "    x264-1!157.20191217        |       h7b6447c_0         922 KB\n",
      "    xz-5.2.5                   |       h7f8727e_1         339 KB\n",
      "    zlib-1.2.12                |       h7f8727e_2         106 KB\n",
      "    ------------------------------------------------------------\n",
      "                                           Total:        89.5 MB\n",
      "\n",
      "The following NEW packages will be INSTALLED:\n",
      "\n",
      "  cytoolz            pkgs/main/linux-64::cytoolz-0.11.0-py37h7b6447c_0\n",
      "  lerc               pkgs/main/linux-64::lerc-3.0-h295c915_0\n",
      "  libdeflate         pkgs/main/linux-64::libdeflate-1.8-h7f8727e_5\n",
      "  libopus            pkgs/main/linux-64::libopus-1.3.1-h7b6447c_0\n",
      "  libvpx             pkgs/main/linux-64::libvpx-1.7.0-h439df22_0\n",
      "  toolz              pkgs/main/noarch::toolz-0.11.2-pyhd3eb1b0_0\n",
      "  x264               pkgs/main/linux-64::x264-1!157.20191217-h7b6447c_0\n",
      "\n",
      "The following packages will be REMOVED:\n",
      "\n",
      "  colorama-0.4.4-pyhd3eb1b0_0\n",
      "  conda-content-trust-0.1.1-pyhd3eb1b0_0\n",
      "  libiconv-1.16-h7f8727e_2\n",
      "  tzdata-2022a-hda174b7_0\n",
      "\n",
      "The following packages will be UPDATED:\n",
      "\n",
      "  _openmp_mutex                                   4.5-1_gnu --> 5.1-1_gnu\n",
      "  ca-certificates                      2022.4.26-h06a4308_0 --> 2022.07.19-h06a4308_0\n",
      "  cffi                                1.15.0-py37hd667e15_1 --> 1.15.1-py37h74dc2b5_0\n",
      "  conda                               4.13.0-py37h06a4308_0 --> 4.14.0-py37h06a4308_0\n",
      "  ld_impl_linux-64                        2.35.1-h7274673_9 --> 2.38-h1181459_1\n",
      "  libgcc-ng                               9.3.0-h5101ec6_17 --> 11.2.0-h1234567_1\n",
      "  libgomp                                 9.3.0-h5101ec6_17 --> 11.2.0-h1234567_1\n",
      "  libstdcxx-ng                            9.3.0-hd4cf53a_17 --> 11.2.0-h1234567_1\n",
      "  libtiff                                  4.2.0-h2818925_1 --> 4.4.0-hecacb30_0\n",
      "  ncurses                                    6.3-h7f8727e_2 --> 6.3-h5eee18b_3\n",
      "  openssl                                 1.1.1o-h7f8727e_0 --> 1.1.1q-h7f8727e_0\n",
      "  pillow                               9.0.1-py37h22f2fdc_0 --> 9.2.0-py37hace64e9_1\n",
      "  pip                                 21.2.2-py37h06a4308_0 --> 22.1.2-py37h06a4308_0\n",
      "  psutil                               5.8.0-py37h27cfd23_1 --> 5.9.0-py37h5eee18b_0\n",
      "  requests           pkgs/main/noarch::requests-2.27.1-pyh~ --> pkgs/main/linux-64::requests-2.28.1-py37h06a4308_0\n",
      "  ripgrep                                          12.1.1-0 --> 13.0.0-hbdeaff8_0\n",
      "  setuptools                          61.2.0-py37h06a4308_0 --> 63.4.1-py37h06a4308_0\n",
      "  sqlite                                  3.38.2-hc218d9a_0 --> 3.39.2-h5082296_0\n",
      "  tk                                      8.6.11-h1ccaba5_0 --> 8.6.12-h1ccaba5_0\n",
      "  tqdm               pkgs/main/noarch::tqdm-4.63.0-pyhd3eb~ --> pkgs/main/linux-64::tqdm-4.64.0-py37h06a4308_0\n",
      "  typing_extensions  pkgs/main/noarch::typing_extensions-4~ --> pkgs/main/linux-64::typing_extensions-4.3.0-py37h06a4308_0\n",
      "  urllib3            pkgs/main/noarch::urllib3-1.26.8-pyhd~ --> pkgs/main/linux-64::urllib3-1.26.11-py37h06a4308_0\n",
      "  xz                                       5.2.5-h7b6447c_0 --> 5.2.5-h7f8727e_1\n",
      "  zlib                                    1.2.12-h7f8727e_1 --> 1.2.12-h7f8727e_2\n",
      "\n",
      "The following packages will be SUPERSEDED by a higher-priority channel:\n",
      "\n",
      "  ffmpeg                     pytorch::ffmpeg-4.3-hf484d3e_0 --> pkgs/main::ffmpeg-4.2.2-h20bf706_0\n",
      "\n",
      "\n",
      "\n",
      "Downloading and Extracting Packages\n",
      "libdeflate-1.8       | 51 KB     | ##################################### | 100% \n",
      "libstdcxx-ng-11.2.0  | 4.7 MB    | ##################################### | 100% \n",
      "typing_extensions-4. | 42 KB     | ##################################### | 100% \n",
      "psutil-5.9.0         | 328 KB    | ##################################### | 100% \n",
      "ld_impl_linux-64-2.3 | 654 KB    | ##################################### | 100% \n",
      "ca-certificates-2022 | 124 KB    | ##################################### | 100% \n",
      "urllib3-1.26.11      | 181 KB    | ##################################### | 100% \n",
      "ripgrep-13.0.0       | 1.1 MB    | ##################################### | 100% \n",
      "openssl-1.1.1q       | 2.5 MB    | ##################################### | 100% \n",
      "x264-1!157.20191217  | 922 KB    | ##################################### | 100% \n",
      "libgcc-ng-11.2.0     | 5.3 MB    | ##################################### | 100% \n",
      "pillow-9.2.0         | 664 KB    | ##################################### | 100% \n",
      "cffi-1.15.1          | 227 KB    | ##################################### | 100% \n",
      "xz-5.2.5             | 339 KB    | ##################################### | 100% \n",
      "cytoolz-0.11.0       | 328 KB    | ##################################### | 100% \n",
      "setuptools-63.4.1    | 1.1 MB    | ##################################### | 100% \n",
      "ffmpeg-4.2.2         | 59.6 MB   | ##################################### | 100% \n",
      "_openmp_mutex-5.1    | 21 KB     | ##################################### | 100% \n",
      "libgomp-11.2.0       | 474 KB    | ##################################### | 100% \n",
      "zlib-1.2.12          | 106 KB    | ##################################### | 100% \n",
      "tqdm-4.64.0          | 126 KB    | ##################################### | 100% \n",
      "requests-2.28.1      | 92 KB     | ##################################### | 100% \n",
      "conda-4.14.0         | 909 KB    | ##################################### | 100% \n",
      "ncurses-6.3          | 781 KB    | ##################################### | 100% \n",
      "lerc-3.0             | 196 KB    | ##################################### | 100% \n",
      "libopus-1.3.1        | 491 KB    | ##################################### | 100% \n",
      "tk-8.6.12            | 3.0 MB    | ##################################### | 100% \n",
      "pip-22.1.2           | 2.4 MB    | ##################################### | 100% \n",
      "libvpx-1.7.0         | 1.2 MB    | ##################################### | 100% \n",
      "libtiff-4.4.0        | 471 KB    | ##################################### | 100% \n",
      "sqlite-3.39.2        | 1.1 MB    | ##################################### | 100% \n",
      "toolz-0.11.2         | 49 KB     | ##################################### | 100% \n",
      "Preparing transaction: done\n",
      "Verifying transaction: done\n",
      "Executing transaction: done\n"
     ]
    }
   ],
   "source": [
    "# Update conda\n",
    "!conda update --force conda -y"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "655873d5-9b30-4c00-89ce-d506b8694877",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Will remove 32 (89.5 MB) tarball(s).\n",
      "Will remove 1 index cache(s).\n",
      "Will remove 1 (103 KB) package(s).\n",
      "There are no tempfile(s) to remove.\n",
      "There are no logfile(s) to remove.\n"
     ]
    }
   ],
   "source": [
    "# clean up cached conda packages from whatever came with the docker container\n",
    "!conda clean --all -y"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "id": "8cb95197-65fb-4cd0-b9da-8a577e7df35b",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Collecting package metadata (current_repodata.json): done\n",
      "Solving environment: done\n",
      "\n",
      "## Package Plan ##\n",
      "\n",
      "  environment location: /workspace/sd_env\n",
      "\n",
      "  added / updated specs:\n",
      "    - python=3.8\n",
      "\n",
      "\n",
      "The following packages will be downloaded:\n",
      "\n",
      "    package                    |            build\n",
      "    ---------------------------|-----------------\n",
      "    _libgcc_mutex-0.1          |             main           3 KB\n",
      "    _openmp_mutex-5.1          |            1_gnu          21 KB\n",
      "    certifi-2022.6.15          |   py38h06a4308_0         153 KB\n",
      "    pip-22.1.2                 |   py38h06a4308_0         2.5 MB\n",
      "    python-3.8.13              |       h12debd9_0        18.8 MB\n",
      "    setuptools-63.4.1          |   py38h06a4308_0         1.1 MB\n",
      "    ------------------------------------------------------------\n",
      "                                           Total:        22.5 MB\n",
      "\n",
      "The following NEW packages will be INSTALLED:\n",
      "\n",
      "  _libgcc_mutex      pkgs/main/linux-64::_libgcc_mutex-0.1-main\n",
      "  _openmp_mutex      pkgs/main/linux-64::_openmp_mutex-5.1-1_gnu\n",
      "  ca-certificates    pkgs/main/linux-64::ca-certificates-2022.07.19-h06a4308_0\n",
      "  certifi            pkgs/main/linux-64::certifi-2022.6.15-py38h06a4308_0\n",
      "  ld_impl_linux-64   pkgs/main/linux-64::ld_impl_linux-64-2.38-h1181459_1\n",
      "  libffi             pkgs/main/linux-64::libffi-3.3-he6710b0_2\n",
      "  libgcc-ng          pkgs/main/linux-64::libgcc-ng-11.2.0-h1234567_1\n",
      "  libgomp            pkgs/main/linux-64::libgomp-11.2.0-h1234567_1\n",
      "  libstdcxx-ng       pkgs/main/linux-64::libstdcxx-ng-11.2.0-h1234567_1\n",
      "  ncurses            pkgs/main/linux-64::ncurses-6.3-h5eee18b_3\n",
      "  openssl            pkgs/main/linux-64::openssl-1.1.1q-h7f8727e_0\n",
      "  pip                pkgs/main/linux-64::pip-22.1.2-py38h06a4308_0\n",
      "  python             pkgs/main/linux-64::python-3.8.13-h12debd9_0\n",
      "  readline           pkgs/main/linux-64::readline-8.1.2-h7f8727e_1\n",
      "  setuptools         pkgs/main/linux-64::setuptools-63.4.1-py38h06a4308_0\n",
      "  sqlite             pkgs/main/linux-64::sqlite-3.39.2-h5082296_0\n",
      "  tk                 pkgs/main/linux-64::tk-8.6.12-h1ccaba5_0\n",
      "  wheel              pkgs/main/noarch::wheel-0.37.1-pyhd3eb1b0_0\n",
      "  xz                 pkgs/main/linux-64::xz-5.2.5-h7f8727e_1\n",
      "  zlib               pkgs/main/linux-64::zlib-1.2.12-h7f8727e_2\n",
      "\n",
      "\n",
      "\n",
      "Downloading and Extracting Packages\n",
      "_libgcc_mutex-0.1    | 3 KB      | ##################################### | 100% \n",
      "setuptools-63.4.1    | 1.1 MB    | ##################################### | 100% \n",
      "python-3.8.13        | 18.8 MB   | ##################################### | 100% \n",
      "_openmp_mutex-5.1    | 21 KB     | ##################################### | 100% \n",
      "pip-22.1.2           | 2.5 MB    | ##################################### | 100% \n",
      "certifi-2022.6.15    | 153 KB    | ##################################### | 100% \n",
      "Preparing transaction: done\n",
      "Verifying transaction: done\n",
      "Executing transaction: done\n",
      "#\n",
      "# To activate this environment, use\n",
      "#\n",
      "#     $ conda activate /workspace/sd_env\n",
      "#\n",
      "# To deactivate an active environment, use\n",
      "#\n",
      "#     $ conda deactivate\n",
      "\n",
      "Retrieving notices: ...working... done\n"
     ]
    }
   ],
   "source": [
    "# create new conda env\n",
    "!mkdir -p {wd}/sd_env\n",
    "!conda create -p {wd}/sd_env/ python=3.8 -y"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "id": "b20d2df3-0f67-444f-95a5-9f151adaab40",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Cloning into 'stable-diffusion'...\n",
      "remote: Enumerating objects: 995, done.\u001b[K\n",
      "remote: Counting objects: 100% (78/78), done.\u001b[K\n",
      "remote: Compressing objects: 100% (43/43), done.\u001b[K\n",
      "remote: Total 995 (delta 30), reused 72 (delta 29), pack-reused 917\u001b[K\n",
      "Receiving objects: 100% (995/995), 42.67 MiB | 26.27 MiB/s, done.\n",
      "Resolving deltas: 100% (480/480), done.\n",
      "/workspace/stable-diffusion\n",
      "Obtaining file:///workspace/stable-diffusion\n",
      "  Preparing metadata (setup.py) ... \u001b[?25ldone\n",
      "\u001b[?25hCollecting torch\n",
      "  Downloading torch-1.12.1-cp38-cp38-manylinux1_x86_64.whl (776.3 MB)\n",
      "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m776.3/776.3 MB\u001b[0m \u001b[31m7.9 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m00:01\u001b[0m00:01\u001b[0m\n",
      "\u001b[?25hCollecting numpy\n",
      "  Downloading numpy-1.23.2-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (17.1 MB)\n",
      "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m17.1/17.1 MB\u001b[0m \u001b[31m86.9 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m00:01\u001b[0m00:01\u001b[0m\n",
      "\u001b[?25hCollecting tqdm\n",
      "  Downloading tqdm-4.64.0-py2.py3-none-any.whl (78 kB)\n",
      "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m78.4/78.4 kB\u001b[0m \u001b[31m54.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
      "\u001b[?25hCollecting typing-extensions\n",
      "  Downloading typing_extensions-4.3.0-py3-none-any.whl (25 kB)\n",
      "Installing collected packages: typing-extensions, tqdm, numpy, torch, latent-diffusion\n",
      "  Running setup.py develop for latent-diffusion\n",
      "Successfully installed latent-diffusion-0.0.1 numpy-1.23.2 torch-1.12.1 tqdm-4.64.0 typing-extensions-4.3.0\n",
      "\u001b[33mWARNING: Running pip as the 'root' user can result in broken permissions and conflicting behaviour with the system package manager. It is recommended to use a virtual environment instead: https://pip.pypa.io/warnings/venv\u001b[0m\u001b[33m\n",
      "\u001b[0m"
     ]
    }
   ],
   "source": [
    "!git clone https://github.com/hlky/stable-diffusion\n",
    "%cd stable-diffusion\n",
    "!{pip_path} install -e ."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "id": "77ac5b33-d46a-47e9-93ba-c805a029c0e9",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "GPU 0: NVIDIA A100 80GB PCIe (UUID: GPU-d9c7c489-f1a8-03ff-52c7-a405cb020992)\n"
     ]
    }
   ],
   "source": [
    "# make sure we didnt break cuda\n",
    "!nvidia-smi -L"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "id": "77d6fc35-092a-420e-8788-773b33ecc06b",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\n",
      "\n",
      "WARNING: The --force flag will be removed in a future conda release.\n",
      "         See 'conda install --help' for details about the --force-reinstall\n",
      "         and --clobber flags.\n",
      "\n",
      "\n",
      "Collecting package metadata (current_repodata.json): done\n",
      "Solving environment: failed with initial frozen solve. Retrying with flexible solve.\n",
      "Collecting package metadata (repodata.json): done\n",
      "Solving environment: done\n",
      "\n",
      "## Package Plan ##\n",
      "\n",
      "  environment location: /workspace/sd_env\n",
      "\n",
      "  added / updated specs:\n",
      "    - pytorch=1.11.0\n",
      "\n",
      "\n",
      "The following packages will be downloaded:\n",
      "\n",
      "    package                    |            build\n",
      "    ---------------------------|-----------------\n",
      "    blas-1.0                   |              mkl           6 KB\n",
      "    cudatoolkit-11.3.1         |       h2bc3f7f_2       549.3 MB\n",
      "    intel-openmp-2022.0.1      |    h06a4308_3633         4.2 MB\n",
      "    libuv-1.40.0               |       h7b6447c_0         736 KB\n",
      "    mkl-2022.0.1               |     h06a4308_117       127.7 MB\n",
      "    pytorch-1.11.0             |py3.8_cuda11.3_cudnn8.2.0_0        1.02 GB  pytorch\n",
      "    pytorch-mutex-1.0          |             cuda           3 KB  pytorch\n",
      "    typing_extensions-4.3.0    |   py38h06a4308_0          42 KB\n",
      "    ------------------------------------------------------------\n",
      "                                           Total:        1.69 GB\n",
      "\n",
      "The following NEW packages will be INSTALLED:\n",
      "\n",
      "  blas               pkgs/main/linux-64::blas-1.0-mkl\n",
      "  cudatoolkit        pkgs/main/linux-64::cudatoolkit-11.3.1-h2bc3f7f_2\n",
      "  intel-openmp       pkgs/main/linux-64::intel-openmp-2022.0.1-h06a4308_3633\n",
      "  libuv              pkgs/main/linux-64::libuv-1.40.0-h7b6447c_0\n",
      "  mkl                pkgs/main/linux-64::mkl-2022.0.1-h06a4308_117\n",
      "  pytorch            pytorch/linux-64::pytorch-1.11.0-py3.8_cuda11.3_cudnn8.2.0_0\n",
      "  pytorch-mutex      pytorch/noarch::pytorch-mutex-1.0-cuda\n",
      "  typing_extensions  pkgs/main/linux-64::typing_extensions-4.3.0-py38h06a4308_0\n",
      "\n",
      "\n",
      "\n",
      "Downloading and Extracting Packages\n",
      "mkl-2022.0.1         | 127.7 MB  | ##################################### | 100% \n",
      "pytorch-mutex-1.0    | 3 KB      | ##################################### | 100% \n",
      "blas-1.0             | 6 KB      | ##################################### | 100% \n",
      "libuv-1.40.0         | 736 KB    | ##################################### | 100% \n",
      "intel-openmp-2022.0. | 4.2 MB    | ##################################### | 100% \n",
      "cudatoolkit-11.3.1   | 549.3 MB  | ##################################### | 100% \n",
      "pytorch-1.11.0       | 1.02 GB   | ##################################### | 100% \n",
      "typing_extensions-4. | 42 KB     | ##################################### | 100% \n",
      "Preparing transaction: done\n",
      "Verifying transaction: done\n",
      "Executing transaction: / By downloading and using the CUDA Toolkit conda packages, you accept the terms and conditions of the CUDA End User License Agreement (EULA): https://docs.nvidia.com/cuda/eula/index.html\n",
      "\n",
      "done\n",
      "Retrieving notices: ...working... done\n"
     ]
    }
   ],
   "source": [
    "# Force the correct pytorch version and dependencies before proceeding. \n",
    "# Incorrect versions of pytorch/torchvision/torchtext will put you into dependency hell\n",
    "!conda install -f -p {conda_env} -c pytorch pytorch=1.11.0 -y"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "id": "9c985413-14ce-4d45-869a-b59867407aa2",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Collecting package metadata (repodata.json): done\n",
      "Solving environment: done\n",
      "\n",
      "Downloading and Extracting Packages\n",
      "pip-20.3.3           | 1.8 MB    | ##################################### | 100% \n",
      "urllib3-1.26.11      | 182 KB    | ##################################### | 100% \n",
      "krb5-1.19.2          | 1.2 MB    | ##################################### | 100% \n",
      "requests-2.28.1      | 92 KB     | ##################################### | 100% \n",
      "cffi-1.15.1          | 228 KB    | ##################################### | 100% \n",
      "cryptography-37.0.1  | 1.3 MB    | ##################################### | 100% \n",
      "python-3.8.5         | 49.3 MB   | ##################################### | 100% \n",
      "torchvision-0.12.0   | 27.7 MB   | ##################################### | 100% \n",
      "mkl-2020.2           | 138.3 MB  | ##################################### | 100% \n",
      "curl-7.84.0          | 79 KB     | ##################################### | 100% \n",
      "mkl_fft-1.3.0        | 177 KB    | ##################################### | 100% \n",
      "pillow-9.2.0         | 666 KB    | ##################################### | 100% \n",
      "libnghttp2-1.46.0    | 680 KB    | ##################################### | 100% \n",
      "libcurl-7.84.0       | 337 KB    | ##################################### | 100% \n",
      "pysocks-1.7.1        | 31 KB     | ##################################### | 100% \n",
      "pcre2-10.37          | 839 KB    | ##################################### | 100% \n",
      "gettext-0.21.0       | 2.6 MB    | ##################################### | 100% \n",
      "c-ares-1.18.1        | 114 KB    | ##################################### | 100% \n",
      "git-2.34.1           | 6.0 MB    | ##################################### | 100% \n",
      "numpy-base-1.19.2    | 4.2 MB    | ##################################### | 100% \n",
      "expat-2.4.4          | 169 KB    | ##################################### | 100% \n",
      "mkl_random-1.1.1     | 341 KB    | ##################################### | 100% \n",
      "perl-5.26.2          | 10.5 MB   | ##################################### | 100% \n",
      "mkl-service-2.3.0    | 62 KB     | ##################################### | 100% \n",
      "libev-4.33           | 111 KB    | ##################################### | 100% \n",
      "brotlipy-0.7.0       | 323 KB    | ##################################### | 100% \n",
      "libssh2-1.10.0       | 274 KB    | ##################################### | 100% \n",
      "numpy-1.19.2         | 22 KB     | ##################################### | 100% \n",
      "libedit-3.1.20210910 | 166 KB    | ##################################### | 100% \n",
      "Preparing transaction: done\n",
      "Verifying transaction: done\n",
      "Executing transaction: done\n",
      "Installing pip dependencies: \\ Ran pip subprocess with arguments:\n",
      "['/workspace/sd_env/bin/python', '-m', 'pip', 'install', '-U', '-r', '/workspace/stable-diffusion/condaenv.znnj49o_.requirements.txt']\n",
      "Pip subprocess output:\n",
      "Obtaining facexlib from git+https://github.com/hlky/facexlib#egg=facexlib (from -r /workspace/stable-diffusion/condaenv.znnj49o_.requirements.txt (line 20))\n",
      "  Cloning https://github.com/hlky/facexlib to ./src/facexlib\n",
      "Obtaining taming-transformers from git+https://github.com/CompVis/taming-transformers#egg=taming-transformers (from -r /workspace/stable-diffusion/condaenv.znnj49o_.requirements.txt (line 21))\n",
      "  Cloning https://github.com/CompVis/taming-transformers to ./src/taming-transformers\n",
      "Obtaining clip from git+https://github.com/openai/CLIP#egg=clip (from -r /workspace/stable-diffusion/condaenv.znnj49o_.requirements.txt (line 22))\n",
      "  Cloning https://github.com/openai/CLIP to ./src/clip\n",
      "Obtaining GFPGAN from git+https://github.com/TencentARC/GFPGAN#egg=GFPGAN (from -r /workspace/stable-diffusion/condaenv.znnj49o_.requirements.txt (line 23))\n",
      "  Cloning https://github.com/TencentARC/GFPGAN to ./src/gfpgan\n",
      "Obtaining realesrgan from git+https://github.com/xinntao/Real-ESRGAN#egg=realesrgan (from -r /workspace/stable-diffusion/condaenv.znnj49o_.requirements.txt (line 24))\n",
      "  Cloning https://github.com/xinntao/Real-ESRGAN to ./src/realesrgan\n",
      "Obtaining k_diffusion from git+https://github.com/hlky/k-diffusion-sd#egg=k_diffusion (from -r /workspace/stable-diffusion/condaenv.znnj49o_.requirements.txt (line 25))\n",
      "  Cloning https://github.com/hlky/k-diffusion-sd to ./src/k-diffusion\n",
      "  Installing build dependencies: started\n",
      "  Installing build dependencies: finished with status 'done'\n",
      "  Getting requirements to build wheel: started\n",
      "  Getting requirements to build wheel: finished with status 'done'\n",
      "  Installing backend dependencies: started\n",
      "  Installing backend dependencies: finished with status 'done'\n",
      "    Preparing wheel metadata: started\n",
      "    Preparing wheel metadata: finished with status 'done'\n",
      "Obtaining file:///workspace/stable-diffusion (from -r /workspace/stable-diffusion/condaenv.znnj49o_.requirements.txt (line 26))\n",
      "Requirement already satisfied: tqdm in /workspace/sd_env/lib/python3.8/site-packages (from clip->-r /workspace/stable-diffusion/condaenv.znnj49o_.requirements.txt (line 22)) (4.64.0)\n",
      "Requirement already satisfied: torch in /workspace/sd_env/lib/python3.8/site-packages (from clip->-r /workspace/stable-diffusion/condaenv.znnj49o_.requirements.txt (line 22)) (1.12.1)\n",
      "Requirement already satisfied: torchvision in /workspace/sd_env/lib/python3.8/site-packages (from clip->-r /workspace/stable-diffusion/condaenv.znnj49o_.requirements.txt (line 22)) (0.12.0)\n",
      "Requirement already satisfied: numpy in /workspace/sd_env/lib/python3.8/site-packages (from facexlib->-r /workspace/stable-diffusion/condaenv.znnj49o_.requirements.txt (line 20)) (1.23.2)\n",
      "Requirement already satisfied: Pillow in /workspace/sd_env/lib/python3.8/site-packages (from facexlib->-r /workspace/stable-diffusion/condaenv.znnj49o_.requirements.txt (line 20)) (9.2.0)\n",
      "Collecting accelerate==0.12.0\n",
      "  Downloading accelerate-0.12.0-py3-none-any.whl (143 kB)\n",
      "Collecting albumentations==0.4.3\n",
      "  Downloading albumentations-0.4.3.tar.gz (3.2 MB)\n",
      "Collecting einops==0.3.0\n",
      "  Downloading einops-0.3.0-py2.py3-none-any.whl (25 kB)\n",
      "Collecting gradio==3.1.6\n",
      "  Downloading gradio-3.1.6-py3-none-any.whl (6.1 MB)\n",
      "Requirement already satisfied: requests in /workspace/sd_env/lib/python3.8/site-packages (from gradio==3.1.6->-r /workspace/stable-diffusion/condaenv.znnj49o_.requirements.txt (line 16)) (2.28.1)\n",
      "Collecting imageio==2.9.0\n",
      "  Downloading imageio-2.9.0-py3-none-any.whl (3.3 MB)\n",
      "Collecting imageio-ffmpeg==0.4.2\n",
      "  Downloading imageio_ffmpeg-0.4.2-py3-none-manylinux2010_x86_64.whl (26.9 MB)\n",
      "Collecting kornia==0.6\n",
      "  Downloading kornia-0.6.0-py2.py3-none-any.whl (367 kB)\n",
      "Collecting omegaconf==2.1.1\n",
      "  Downloading omegaconf-2.1.1-py3-none-any.whl (74 kB)\n",
      "Collecting opencv-python==4.1.2.30\n",
      "  Downloading opencv_python-4.1.2.30-cp38-cp38-manylinux1_x86_64.whl (28.3 MB)\n",
      "Collecting opencv-python-headless==4.1.2.30\n",
      "  Downloading opencv_python_headless-4.1.2.30-cp38-cp38-manylinux1_x86_64.whl (21.8 MB)\n",
      "Collecting pudb==2019.2\n",
      "  Downloading pudb-2019.2.tar.gz (59 kB)\n",
      "Collecting pynvml==11.4.1\n",
      "  Downloading pynvml-11.4.1-py3-none-any.whl (46 kB)\n",
      "Collecting pytorch-lightning==1.4.2\n",
      "  Downloading pytorch_lightning-1.4.2-py3-none-any.whl (916 kB)\n",
      "Requirement already satisfied: typing-extensions in /workspace/sd_env/lib/python3.8/site-packages (from pytorch-lightning==1.4.2->-r /workspace/stable-diffusion/condaenv.znnj49o_.requirements.txt (line 7)) (4.3.0)\n",
      "Collecting torch-fidelity==0.3.0\n",
      "  Downloading torch_fidelity-0.3.0-py3-none-any.whl (37 kB)\n",
      "Collecting torchmetrics==0.6.0\n",
      "  Downloading torchmetrics-0.6.0-py3-none-any.whl (329 kB)\n",
      "Collecting transformers==4.19.2\n",
      "  Downloading transformers-4.19.2-py3-none-any.whl (4.2 MB)\n",
      "Collecting antlr4-python3-runtime==4.8\n",
      "  Downloading antlr4-python3-runtime-4.8.tar.gz (112 kB)\n",
      "Collecting pyDeprecate==0.3.1\n",
      "  Downloading pyDeprecate-0.3.1-py3-none-any.whl (10 kB)\n",
      "Collecting basicsr>=1.3.4.0\n",
      "  Downloading basicsr-1.4.2.tar.gz (172 kB)\n",
      "Collecting streamlit>=0.73.1\n",
      "  Downloading streamlit-1.12.2-py2.py3-none-any.whl (9.1 MB)\n",
      "Collecting test-tube>=0.7.5\n",
      "  Downloading test_tube-0.7.5.tar.gz (21 kB)\n",
      "Collecting altair>=3.2.0\n",
      "  Downloading altair-4.2.0-py3-none-any.whl (812 kB)\n",
      "Collecting blinker>=1.0.0\n",
      "  Downloading blinker-1.5-py2.py3-none-any.whl (12 kB)\n",
      "Collecting cachetools>=4.0\n",
      "  Downloading cachetools-5.2.0-py3-none-any.whl (9.3 kB)\n",
      "Collecting click>=7.0\n",
      "  Downloading click-8.1.3-py3-none-any.whl (96 kB)\n",
      "Collecting fsspec\n",
      "  Downloading fsspec-2022.8.0-py3-none-any.whl (140 kB)\n",
      "Collecting aiohttp\n",
      "  Downloading aiohttp-3.8.1-cp38-cp38-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_12_x86_64.manylinux2010_x86_64.whl (1.3 MB)\n",
      "Requirement already satisfied: charset-normalizer<3.0,>=2.0 in /workspace/sd_env/lib/python3.8/site-packages (from aiohttp->gradio==3.1.6->-r /workspace/stable-diffusion/condaenv.znnj49o_.requirements.txt (line 16)) (2.0.4)\n",
      "Collecting aiosignal>=1.1.2\n",
      "  Downloading aiosignal-1.2.0-py3-none-any.whl (8.2 kB)\n",
      "Collecting async-timeout<5.0,>=4.0.0a3\n",
      "  Downloading async_timeout-4.0.2-py3-none-any.whl (5.8 kB)\n",
      "Collecting attrs>=17.3.0\n",
      "  Using cached attrs-22.1.0-py2.py3-none-any.whl (58 kB)\n",
      "Collecting frozenlist>=1.1.1\n",
      "  Downloading frozenlist-1.3.1-cp38-cp38-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_17_x86_64.manylinux2014_x86_64.whl (161 kB)\n",
      "Collecting future>=0.17.1\n",
      "  Downloading future-0.18.2.tar.gz (829 kB)\n",
      "Collecting gitpython!=3.1.19\n",
      "  Downloading GitPython-3.1.27-py3-none-any.whl (181 kB)\n",
      "Collecting gitdb<5,>=4.0.1\n",
      "  Downloading gitdb-4.0.9-py3-none-any.whl (63 kB)\n",
      "Collecting h11<0.13,>=0.11\n",
      "  Downloading h11-0.12.0-py3-none-any.whl (54 kB)\n",
      "Collecting huggingface-hub<1.0,>=0.1.0\n",
      "  Downloading huggingface_hub-0.9.1-py3-none-any.whl (120 kB)\n",
      "Collecting imgaug<0.2.7,>=0.2.5\n",
      "  Downloading imgaug-0.2.6.tar.gz (631 kB)\n",
      "Requirement already satisfied: six in /workspace/sd_env/lib/python3.8/site-packages (from imgaug<0.2.7,>=0.2.5->albumentations==0.4.3->-r /workspace/stable-diffusion/condaenv.znnj49o_.requirements.txt (line 1)) (1.16.0)\n",
      "Collecting importlib-metadata>=1.4\n",
      "  Using cached importlib_metadata-4.12.0-py3-none-any.whl (21 kB)\n",
      "Collecting jsonschema>=3.0\n",
      "  Downloading jsonschema-4.14.0-py3-none-any.whl (82 kB)\n",
      "Collecting importlib-resources>=1.4.0\n",
      "  Using cached importlib_resources-5.9.0-py3-none-any.whl (33 kB)\n",
      "Collecting multidict<7.0,>=4.5\n",
      "  Downloading multidict-6.0.2-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (121 kB)\n",
      "Collecting numpy\n",
      "  Downloading numpy-1.20.3-cp38-cp38-manylinux_2_12_x86_64.manylinux2010_x86_64.whl (15.4 MB)\n",
      "Collecting packaging>=20.0\n",
      "  Using cached packaging-21.3-py3-none-any.whl (40 kB)\n",
      "Collecting pandas\n",
      "  Downloading pandas-1.4.3-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (11.7 MB)\n",
      "Collecting pkgutil-resolve-name>=1.3.10\n",
      "  Using cached pkgutil_resolve_name-1.3.10-py3-none-any.whl (4.7 kB)\n",
      "Collecting protobuf<4,>=3.12\n",
      "  Downloading protobuf-3.20.1-cp38-cp38-manylinux_2_5_x86_64.manylinux1_x86_64.whl (1.0 MB)\n",
      "Collecting pyarrow>=4.0\n",
      "  Downloading pyarrow-9.0.0-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (35.3 MB)\n",
      "Collecting pydeck>=0.1.dev5\n",
      "  Downloading pydeck-0.8.0b1-py2.py3-none-any.whl (4.7 MB)\n",
      "Collecting Jinja2\n",
      "  Using cached Jinja2-3.1.2-py3-none-any.whl (133 kB)\n",
      "Collecting MarkupSafe>=2.0\n",
      "  Downloading MarkupSafe-2.1.1-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (25 kB)\n",
      "Collecting pygments>=1.0\n",
      "  Downloading Pygments-2.13.0-py3-none-any.whl (1.1 MB)\n",
      "Collecting pympler>=0.9\n",
      "  Downloading Pympler-1.0.1-py3-none-any.whl (164 kB)\n",
      "Collecting pyparsing!=3.0.5,>=2.0.2\n",
      "  Using cached pyparsing-3.0.9-py3-none-any.whl (98 kB)\n",
      "Collecting pyrsistent!=0.17.0,!=0.17.1,!=0.17.2,>=0.14.0\n",
      "  Downloading pyrsistent-0.18.1-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (119 kB)\n",
      "Collecting python-dateutil\n",
      "  Using cached python_dateutil-2.8.2-py2.py3-none-any.whl (247 kB)\n",
      "Collecting pytz>=2020.1\n",
      "  Downloading pytz-2022.2.1-py2.py3-none-any.whl (500 kB)\n",
      "Collecting pyyaml\n",
      "  Downloading PyYAML-6.0-cp38-cp38-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_12_x86_64.manylinux2010_x86_64.whl (701 kB)\n",
      "Collecting regex\n",
      "  Downloading regex-2022.8.17-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (768 kB)\n",
      "Requirement already satisfied: certifi>=2017.4.17 in /workspace/sd_env/lib/python3.8/site-packages (from requests->gradio==3.1.6->-r /workspace/stable-diffusion/condaenv.znnj49o_.requirements.txt (line 16)) (2022.6.15)\n",
      "Requirement already satisfied: idna<4,>=2.5 in /workspace/sd_env/lib/python3.8/site-packages (from requests->gradio==3.1.6->-r /workspace/stable-diffusion/condaenv.znnj49o_.requirements.txt (line 16)) (3.3)\n",
      "Requirement already satisfied: urllib3<1.27,>=1.21.1 in /workspace/sd_env/lib/python3.8/site-packages (from requests->gradio==3.1.6->-r /workspace/stable-diffusion/condaenv.znnj49o_.requirements.txt (line 16)) (1.26.11)\n",
      "Collecting rich>=10.11.0\n",
      "  Downloading rich-12.5.1-py3-none-any.whl (235 kB)\n",
      "Collecting commonmark<0.10.0,>=0.9.0\n",
      "  Downloading commonmark-0.9.1-py2.py3-none-any.whl (51 kB)\n",
      "Collecting scikit-image\n",
      "  Downloading scikit_image-0.19.3-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (14.0 MB)\n",
      "Collecting networkx>=2.2\n",
      "  Downloading networkx-2.8.6-py3-none-any.whl (2.0 MB)\n",
      "Collecting PyWavelets>=1.1.1\n",
      "  Downloading PyWavelets-1.3.0-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (6.9 MB)\n",
      "Collecting scipy\n",
      "  Downloading scipy-1.9.1-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (43.4 MB)\n",
      "Collecting smmap<6,>=3.0.1\n",
      "  Downloading smmap-5.0.0-py3-none-any.whl (24 kB)\n",
      "Collecting tensorboard>=2.2.0\n",
      "  Downloading tensorboard-2.10.0-py3-none-any.whl (5.9 MB)\n",
      "Requirement already satisfied: setuptools>=41.0.0 in /workspace/sd_env/lib/python3.8/site-packages (from tensorboard>=2.2.0->pytorch-lightning==1.4.2->-r /workspace/stable-diffusion/condaenv.znnj49o_.requirements.txt (line 7)) (63.4.1)\n",
      "Requirement already satisfied: wheel>=0.26 in /workspace/sd_env/lib/python3.8/site-packages (from tensorboard>=2.2.0->pytorch-lightning==1.4.2->-r /workspace/stable-diffusion/condaenv.znnj49o_.requirements.txt (line 7)) (0.37.1)\n",
      "Collecting absl-py>=0.4\n",
      "  Downloading absl_py-1.2.0-py3-none-any.whl (123 kB)\n",
      "Collecting google-auth<3,>=1.6.3\n",
      "  Downloading google_auth-2.11.0-py2.py3-none-any.whl (167 kB)\n",
      "Collecting google-auth-oauthlib<0.5,>=0.4.1\n",
      "  Downloading google_auth_oauthlib-0.4.6-py2.py3-none-any.whl (18 kB)\n",
      "Collecting grpcio>=1.24.3\n",
      "  Downloading grpcio-1.47.0-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (4.5 MB)\n",
      "Collecting markdown>=2.6.8\n",
      "  Downloading Markdown-3.4.1-py3-none-any.whl (93 kB)\n",
      "Collecting protobuf<4,>=3.12\n",
      "  Downloading protobuf-3.19.4-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (1.1 MB)\n",
      "Collecting pyasn1-modules>=0.2.1\n",
      "  Downloading pyasn1_modules-0.2.8-py2.py3-none-any.whl (155 kB)\n",
      "Collecting pyasn1<0.5.0,>=0.4.6\n",
      "  Downloading pyasn1-0.4.8-py2.py3-none-any.whl (77 kB)\n",
      "Collecting requests-oauthlib>=0.7.0\n",
      "  Downloading requests_oauthlib-1.3.1-py2.py3-none-any.whl (23 kB)\n",
      "Collecting oauthlib>=3.0.0\n",
      "  Downloading oauthlib-3.2.0-py3-none-any.whl (151 kB)\n",
      "Collecting rsa<5,>=3.1.4\n",
      "  Downloading rsa-4.9-py3-none-any.whl (34 kB)\n",
      "Collecting tensorboard-data-server<0.7.0,>=0.6.0\n",
      "  Downloading tensorboard_data_server-0.6.1-py3-none-manylinux2010_x86_64.whl (4.9 MB)\n",
      "Collecting tensorboard-plugin-wit>=1.6.0\n",
      "  Downloading tensorboard_plugin_wit-1.8.1-py3-none-any.whl (781 kB)\n",
      "Collecting tifffile>=2019.7.26\n",
      "  Downloading tifffile-2022.8.12-py3-none-any.whl (208 kB)\n",
      "Collecting tokenizers!=0.11.3,<0.13,>=0.11.1\n",
      "  Downloading tokenizers-0.12.1-cp38-cp38-manylinux_2_12_x86_64.manylinux2010_x86_64.whl (6.6 MB)\n",
      "Collecting tornado>=5.0\n",
      "  Using cached tornado-6.2-cp37-abi3-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_17_x86_64.manylinux2014_x86_64.whl (423 kB)\n",
      "Collecting tzlocal>=1.1\n",
      "  Downloading tzlocal-4.2-py3-none-any.whl (19 kB)\n",
      "Collecting urwid>=1.1.1\n",
      "  Downloading urwid-2.1.2.tar.gz (634 kB)\n",
      "Collecting validators>=0.2\n",
      "  Downloading validators-0.20.0.tar.gz (30 kB)\n",
      "Collecting decorator>=3.4.0\n",
      "  Downloading decorator-5.1.1-py3-none-any.whl (9.1 kB)\n",
      "Collecting werkzeug>=1.0.1\n",
      "  Downloading Werkzeug-2.2.2-py3-none-any.whl (232 kB)\n",
      "Collecting yarl<2.0,>=1.0\n",
      "  Downloading yarl-1.8.1-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (262 kB)\n",
      "Collecting zipp>=0.5\n",
      "  Using cached zipp-3.8.1-py3-none-any.whl (5.6 kB)\n",
      "Collecting addict\n",
      "  Downloading addict-2.4.0-py3-none-any.whl (3.8 kB)\n",
      "Collecting analytics-python\n",
      "  Downloading analytics_python-1.4.0-py2.py3-none-any.whl (15 kB)\n",
      "Collecting backoff==1.10.0\n",
      "  Downloading backoff-1.10.0-py2.py3-none-any.whl (31 kB)\n",
      "Collecting monotonic>=1.5\n",
      "  Downloading monotonic-1.6-py2.py3-none-any.whl (8.2 kB)\n",
      "Collecting backports.zoneinfo\n",
      "  Downloading backports.zoneinfo-0.2.1-cp38-cp38-manylinux1_x86_64.whl (74 kB)\n",
      "Collecting clean-fid\n",
      "  Downloading clean_fid-0.1.28-py3-none-any.whl (23 kB)\n",
      "Collecting requests\n",
      "  Downloading requests-2.25.1-py2.py3-none-any.whl (61 kB)\n",
      "Collecting chardet<5,>=3.0.2\n",
      "  Downloading chardet-4.0.0-py2.py3-none-any.whl (178 kB)\n",
      "Collecting idna<4,>=2.5\n",
      "  Downloading idna-2.10-py2.py3-none-any.whl (58 kB)\n",
      "Collecting entrypoints\n",
      "  Using cached entrypoints-0.4-py3-none-any.whl (5.3 kB)\n",
      "Collecting fastapi\n",
      "  Downloading fastapi-0.81.0-py3-none-any.whl (54 kB)\n",
      "Collecting starlette==0.19.1\n",
      "  Downloading starlette-0.19.1-py3-none-any.whl (63 kB)\n",
      "Collecting anyio<5,>=3.4.0\n",
      "  Using cached anyio-3.6.1-py3-none-any.whl (80 kB)\n",
      "Collecting pydantic\n",
      "  Downloading pydantic-1.10.0-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (13.5 MB)\n",
      "Collecting sniffio>=1.1\n",
      "  Using cached sniffio-1.2.0-py3-none-any.whl (10 kB)\n",
      "Collecting ffmpy\n",
      "  Downloading ffmpy-0.3.0.tar.gz (4.8 kB)\n",
      "Collecting filelock\n",
      "  Downloading filelock-3.8.0-py3-none-any.whl (10 kB)\n",
      "Collecting filterpy\n",
      "  Downloading filterpy-1.4.5.zip (177 kB)\n",
      "Collecting ftfy\n",
      "  Downloading ftfy-6.1.1-py3-none-any.whl (53 kB)\n",
      "Collecting wcwidth>=0.2.5\n",
      "  Downloading wcwidth-0.2.5-py2.py3-none-any.whl (30 kB)\n",
      "Collecting httpx\n",
      "  Downloading httpx-0.23.0-py3-none-any.whl (84 kB)\n",
      "Collecting httpcore<0.16.0,>=0.15.0\n",
      "  Downloading httpcore-0.15.0-py3-none-any.whl (68 kB)\n",
      "Collecting rfc3986[idna2008]<2,>=1.3\n",
      "  Downloading rfc3986-1.5.0-py2.py3-none-any.whl (31 kB)\n",
      "Collecting jsonmerge\n",
      "  Downloading jsonmerge-1.8.0.tar.gz (26 kB)\n",
      "Collecting lmdb\n",
      "  Downloading lmdb-1.3.0-cp38-cp38-manylinux_2_12_x86_64.manylinux2010_x86_64.whl (305 kB)\n",
      "Collecting markdown-it-py[linkify,plugins]\n",
      "  Downloading markdown_it_py-2.1.0-py3-none-any.whl (84 kB)\n",
      "Collecting linkify-it-py~=1.0\n",
      "  Downloading linkify_it_py-1.0.3-py3-none-any.whl (19 kB)\n",
      "Collecting mdurl~=0.1\n",
      "  Downloading mdurl-0.1.2-py3-none-any.whl (10.0 kB)\n",
      "Collecting matplotlib\n",
      "  Downloading matplotlib-3.5.3-cp38-cp38-manylinux_2_5_x86_64.manylinux1_x86_64.whl (11.3 MB)\n",
      "Collecting cycler>=0.10\n",
      "  Downloading cycler-0.11.0-py3-none-any.whl (6.4 kB)\n",
      "Collecting fonttools>=4.22.0\n",
      "  Downloading fonttools-4.37.1-py3-none-any.whl (957 kB)\n",
      "Collecting kiwisolver>=1.0.1\n",
      "  Downloading kiwisolver-1.4.4-cp38-cp38-manylinux_2_5_x86_64.manylinux1_x86_64.whl (1.2 MB)\n",
      "Collecting mdit-py-plugins\n",
      "  Downloading mdit_py_plugins-0.3.0-py3-none-any.whl (43 kB)\n",
      "Collecting numba\n",
      "  Downloading numba-0.56.0-cp38-cp38-manylinux2014_x86_64.manylinux_2_17_x86_64.whl (3.5 MB)\n",
      "Collecting llvmlite<0.40,>=0.39.0dev0\n",
      "  Downloading llvmlite-0.39.0-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (34.6 MB)\n",
      "Collecting orjson\n",
      "  Downloading orjson-3.8.0-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (270 kB)\n",
      "Collecting paramiko\n",
      "  Downloading paramiko-2.11.0-py2.py3-none-any.whl (212 kB)\n",
      "Requirement already satisfied: cryptography>=2.5 in /workspace/sd_env/lib/python3.8/site-packages (from paramiko->gradio==3.1.6->-r /workspace/stable-diffusion/condaenv.znnj49o_.requirements.txt (line 16)) (37.0.1)\n",
      "Collecting bcrypt>=3.1.3\n",
      "  Downloading bcrypt-4.0.0-cp36-abi3-manylinux_2_24_x86_64.whl (594 kB)\n",
      "Requirement already satisfied: cffi>=1.12 in /workspace/sd_env/lib/python3.8/site-packages (from cryptography>=2.5->paramiko->gradio==3.1.6->-r /workspace/stable-diffusion/condaenv.znnj49o_.requirements.txt (line 16)) (1.15.1)\n",
      "Requirement already satisfied: pycparser in /workspace/sd_env/lib/python3.8/site-packages (from cffi>=1.12->cryptography>=2.5->paramiko->gradio==3.1.6->-r /workspace/stable-diffusion/condaenv.znnj49o_.requirements.txt (line 16)) (2.21)\n",
      "Collecting pynacl>=1.0.1\n",
      "  Downloading PyNaCl-1.5.0-cp36-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.manylinux_2_24_x86_64.whl (856 kB)\n",
      "Collecting psutil\n",
      "  Downloading psutil-5.9.1-cp38-cp38-manylinux_2_12_x86_64.manylinux2010_x86_64.manylinux_2_17_x86_64.manylinux2014_x86_64.whl (284 kB)\n",
      "Collecting pycryptodome\n",
      "  Downloading pycryptodome-3.15.0-cp35-abi3-manylinux2010_x86_64.whl (2.3 MB)\n",
      "Collecting pydub\n",
      "  Downloading pydub-0.25.1-py2.py3-none-any.whl (32 kB)\n",
      "Collecting python-multipart\n",
      "  Downloading python-multipart-0.0.5.tar.gz (32 kB)\n",
      "Collecting pytz-deprecation-shim\n",
      "  Downloading pytz_deprecation_shim-0.1.0.post0-py2.py3-none-any.whl (15 kB)\n",
      "Collecting resize-right\n",
      "  Downloading resize_right-0.0.2-py3-none-any.whl (8.9 kB)\n",
      "Collecting semver\n",
      "  Downloading semver-2.13.0-py2.py3-none-any.whl (12 kB)\n",
      "Collecting tb-nightly\n",
      "  Downloading tb_nightly-2.11.0a20220830-py3-none-any.whl (5.9 MB)\n",
      "Collecting toml\n",
      "  Downloading toml-0.10.2-py2.py3-none-any.whl (16 kB)\n",
      "Collecting toolz\n",
      "  Downloading toolz-0.12.0-py3-none-any.whl (55 kB)\n",
      "Collecting torchdiffeq\n",
      "  Downloading torchdiffeq-0.2.3-py3-none-any.whl (31 kB)\n",
      "Collecting tzdata\n",
      "  Downloading tzdata-2022.2-py2.py3-none-any.whl (336 kB)\n",
      "Collecting uc-micro-py\n",
      "  Downloading uc_micro_py-1.0.1-py3-none-any.whl (6.2 kB)\n",
      "Collecting uvicorn\n",
      "  Downloading uvicorn-0.18.3-py3-none-any.whl (57 kB)\n",
      "Collecting wandb\n",
      "  Downloading wandb-0.13.2-py2.py3-none-any.whl (1.8 MB)\n",
      "Collecting docker-pycreds>=0.4.0\n",
      "  Downloading docker_pycreds-0.4.0-py2.py3-none-any.whl (9.0 kB)\n",
      "Collecting promise<3,>=2.0\n",
      "  Downloading promise-2.3.tar.gz (19 kB)\n",
      "Collecting sentry-sdk>=1.0.0\n",
      "  Downloading sentry_sdk-1.9.5-py2.py3-none-any.whl (157 kB)\n",
      "Collecting shortuuid>=0.5.0\n",
      "  Downloading shortuuid-1.0.9-py3-none-any.whl (9.4 kB)\n",
      "Collecting pathtools\n",
      "  Downloading pathtools-0.1.2.tar.gz (11 kB)\n",
      "Collecting setproctitle\n",
      "  Downloading setproctitle-1.3.2-cp38-cp38-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_17_x86_64.manylinux2014_x86_64.whl (31 kB)\n",
      "Collecting watchdog\n",
      "  Downloading watchdog-2.1.9-py3-none-manylinux2014_x86_64.whl (78 kB)\n",
      "Collecting websockets\n",
      "  Downloading websockets-10.3-cp38-cp38-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_12_x86_64.manylinux2010_x86_64.whl (111 kB)\n",
      "Collecting yapf\n",
      "  Downloading yapf-0.32.0-py2.py3-none-any.whl (190 kB)\n",
      "Building wheels for collected packages: albumentations, pudb, antlr4-python3-runtime, basicsr, test-tube, future, imgaug, urwid, validators, ffmpy, filterpy, jsonmerge, python-multipart, promise, pathtools\n",
      "  Building wheel for albumentations (setup.py): started\n",
      "  Building wheel for albumentations (setup.py): finished with status 'done'\n",
      "  Created wheel for albumentations: filename=albumentations-0.4.3-py3-none-any.whl size=60767 sha256=6f6a854a19540e84a5f0028223bb28d64730b89bf8dd6511f652c41164222286\n",
      "  Stored in directory: /root/.cache/pip/wheels/a0/37/4e/0bd417ba6a58f73329b825623d8c949e8e4ac2cdbd252b786d\n",
      "  Building wheel for pudb (setup.py): started\n",
      "  Building wheel for pudb (setup.py): finished with status 'done'\n",
      "  Created wheel for pudb: filename=pudb-2019.2-py3-none-any.whl size=63218 sha256=386a606fa3b7c89d46e793d0b898b89865605a61cdf28e55c1c9ff4e2ca14e13\n",
      "  Stored in directory: /root/.cache/pip/wheels/48/83/f1/d8a09d401e2512bfda01ac9fc1b334885f9ddf51617c1c49f1\n",
      "  Building wheel for antlr4-python3-runtime (setup.py): started\n",
      "  Building wheel for antlr4-python3-runtime (setup.py): finished with status 'done'\n",
      "  Created wheel for antlr4-python3-runtime: filename=antlr4_python3_runtime-4.8-py3-none-any.whl size=141210 sha256=1f53628f0b62afd03530505898b2fa62f21cef33d8de9c0962e2bce6a16305c7\n",
      "  Stored in directory: /root/.cache/pip/wheels/c8/d0/ab/d43c02eaddc5b9004db86950802442ad9a26f279c619e28da0\n",
      "  Building wheel for basicsr (setup.py): started\n",
      "  Building wheel for basicsr (setup.py): finished with status 'done'\n",
      "  Created wheel for basicsr: filename=basicsr-1.4.2-py3-none-any.whl size=214823 sha256=17524ddaca15c9c58a78391b357c013aedba69a87c5c16efdd59ab318a435731\n",
      "  Stored in directory: /root/.cache/pip/wheels/4d/d3/95/e17d0bcdd7dcfb0dbf79db006711e434c42036efbf6695ef7f\n",
      "  Building wheel for test-tube (setup.py): started\n",
      "  Building wheel for test-tube (setup.py): finished with status 'done'\n",
      "  Created wheel for test-tube: filename=test_tube-0.7.5-py3-none-any.whl size=25328 sha256=88d589c868984b1b2d8ae3d51830d2661b6d797fac0b187f7ef9ad9f4ef947b2\n",
      "  Stored in directory: /root/.cache/pip/wheels/95/b0/3a/00ea66dbb0d9ce470ce1bdcb854a6fa030c279c316cb27ca9e\n",
      "  Building wheel for future (setup.py): started\n",
      "  Building wheel for future (setup.py): finished with status 'done'\n",
      "  Created wheel for future: filename=future-0.18.2-py3-none-any.whl size=491058 sha256=d1acec1059e0ceecaf0e8b6538a87c40090aec21448465db69ad5f1d7a570518\n",
      "  Stored in directory: /root/.cache/pip/wheels/8e/70/28/3d6ccd6e315f65f245da085482a2e1c7d14b90b30f239e2cf4\n",
      "  Building wheel for imgaug (setup.py): started\n",
      "  Building wheel for imgaug (setup.py): finished with status 'done'\n",
      "  Created wheel for imgaug: filename=imgaug-0.2.6-py3-none-any.whl size=654004 sha256=ff4e69a6a707e757beadb0254fbf1ac29b6b68cb666a5dba1b8752e3ed867cb2\n",
      "  Stored in directory: /root/.cache/pip/wheels/41/23/e8/b1016c275f713978d312621da3c4f55920ec4297798aba8a5a\n",
      "  Building wheel for urwid (setup.py): started\n",
      "  Building wheel for urwid (setup.py): finished with status 'done'\n",
      "  Created wheel for urwid: filename=urwid-2.1.2-py3-none-any.whl size=235033 sha256=6122c07915d8e3741f2e4eabd5471585daf0d73166b93d17b4b554424736c05a\n",
      "  Stored in directory: /root/.cache/pip/wheels/28/71/e4/38b5d81438105d0e3db5016cf2eea6fa796d89d96a04451d4d\n",
      "  Building wheel for validators (setup.py): started\n",
      "  Building wheel for validators (setup.py): finished with status 'done'\n",
      "  Created wheel for validators: filename=validators-0.20.0-py3-none-any.whl size=19579 sha256=2bd8699a29de768da30800ee65c4800dbbb7ffb8316aa76c11e74de7da0058e2\n",
      "  Stored in directory: /root/.cache/pip/wheels/19/09/72/3eb74d236bb48bd0f3c6c3c83e4e0c5bbfcbcad7c6c3539db8\n",
      "  Building wheel for ffmpy (setup.py): started\n",
      "  Building wheel for ffmpy (setup.py): finished with status 'done'\n",
      "  Created wheel for ffmpy: filename=ffmpy-0.3.0-py3-none-any.whl size=4693 sha256=ffbbddb99cd578773158ec56b219c29ea21e3b48c96ce5b914191e86df701a1b\n",
      "  Stored in directory: /root/.cache/pip/wheels/ff/5b/59/913b443e7369dc04b61f607a746b6f7d83fb65e2e19fcc958d\n",
      "  Building wheel for filterpy (setup.py): started\n",
      "  Building wheel for filterpy (setup.py): finished with status 'done'\n",
      "  Created wheel for filterpy: filename=filterpy-1.4.5-py3-none-any.whl size=110458 sha256=4b54e5255fe3d28c809869659dcca7d968258a10fdb8e94f72e4e67083947a35\n",
      "  Stored in directory: /root/.cache/pip/wheels/fe/f6/cb/40331472edf4fd399b8cad02973c6acbdf26898342928327fe\n",
      "  Building wheel for jsonmerge (setup.py): started\n",
      "  Building wheel for jsonmerge (setup.py): finished with status 'done'\n",
      "  Created wheel for jsonmerge: filename=jsonmerge-1.8.0-py3-none-any.whl size=17987 sha256=d2cdfb3ef3236f74f1b53e220775a6182abebbca3684a34d217ff7c9b30a6218\n",
      "  Stored in directory: /root/.cache/pip/wheels/b7/2f/9f/e0ef579cf9c1c669c5f9c18acf76c2b079e9a8ee92c85884e4\n",
      "  Building wheel for python-multipart (setup.py): started\n",
      "  Building wheel for python-multipart (setup.py): finished with status 'done'\n",
      "  Created wheel for python-multipart: filename=python_multipart-0.0.5-py3-none-any.whl size=31671 sha256=b35acb546bae9dfcc29c6b1e0af098d2d82b77348ee490598dca99690b8aeab8\n",
      "  Stored in directory: /root/.cache/pip/wheels/9e/fc/1c/cf980e6413d3ee8e70cd8f39e2366b0f487e3e221aeb452eb0\n",
      "  Building wheel for promise (setup.py): started\n",
      "  Building wheel for promise (setup.py): finished with status 'done'\n",
      "  Created wheel for promise: filename=promise-2.3-py3-none-any.whl size=21486 sha256=ab5edef8e2aa4e645243fb2f2ef9d811d1ae9ed4bd16f530c3b5edc628641e1b\n",
      "  Stored in directory: /root/.cache/pip/wheels/54/aa/01/724885182f93150035a2a91bce34a12877e8067a97baaf5dc8\n",
      "  Building wheel for pathtools (setup.py): started\n",
      "  Building wheel for pathtools (setup.py): finished with status 'done'\n",
      "  Created wheel for pathtools: filename=pathtools-0.1.2-py3-none-any.whl size=8792 sha256=fac124d7ceee0f0fa5d764085fe80eabb8cb4b3b5999210774458d87f15ca986\n",
      "  Stored in directory: /root/.cache/pip/wheels/4c/8e/7e/72fbc243e1aeecae64a96875432e70d4e92f3d2d18123be004\n",
      "Successfully built albumentations pudb antlr4-python3-runtime basicsr test-tube future imgaug urwid validators ffmpy filterpy jsonmerge python-multipart promise pathtools\n",
      "Installing collected packages: pyasn1, idna, chardet, zipp, rsa, requests, pyparsing, pyasn1-modules, oauthlib, cachetools, requests-oauthlib, python-dateutil, packaging, numpy, MarkupSafe, kiwisolver, importlib-metadata, google-auth, fonttools, cycler, werkzeug, tifffile, tensorboard-plugin-wit, tensorboard-data-server, sniffio, smmap, scipy, PyWavelets, protobuf, networkx, multidict, mdurl, matplotlib, markdown, llvmlite, imageio, grpcio, google-auth-oauthlib, frozenlist, absl-py, yarl, yapf, uc-micro-py, tzdata, tb-nightly, scikit-image, rfc3986, pyyaml, pytz, pyrsistent, pkgutil-resolve-name, opencv-python, numba, markdown-it-py, lmdb, importlib-resources, h11, gitdb, future, filterpy, backports.zoneinfo, attrs, async-timeout, anyio, aiosignal, addict, wcwidth, toolz, starlette, shortuuid, setproctitle, sentry-sdk, pytz-deprecation-shim, pynacl, pygments, pydantic, psutil, promise, pathtools, pandas, monotonic, mdit-py-plugins, linkify-it-py, jsonschema, Jinja2, httpcore, gitpython, fsspec, filelock, facexlib, entrypoints, docker-pycreds, decorator, commonmark, click, bcrypt, basicsr, backoff, aiohttp, websockets, watchdog, wandb, validators, uvicorn, urwid, tzlocal, tornado, torchmetrics, torchdiffeq, toml, tokenizers, tensorboard, semver, rich, resize-right, regex, python-multipart, pympler, pydub, pyDeprecate, pydeck, pycryptodome, pyarrow, paramiko, orjson, opencv-python-headless, kornia, jsonmerge, imgaug, huggingface-hub, httpx, GFPGAN, ftfy, ffmpy, fastapi, einops, clean-fid, blinker, antlr4-python3-runtime, analytics-python, altair, accelerate, transformers, torch-fidelity, test-tube, taming-transformers, streamlit, realesrgan, pytorch-lightning, pynvml, pudb, omegaconf, latent-diffusion, k-diffusion, imageio-ffmpeg, gradio, clip, albumentations\n",
      "  Attempting uninstall: idna\n",
      "    Found existing installation: idna 3.3\n",
      "    Uninstalling idna-3.3:\n",
      "      Successfully uninstalled idna-3.3\n",
      "  Attempting uninstall: requests\n",
      "    Found existing installation: requests 2.28.1\n",
      "    Uninstalling requests-2.28.1:\n",
      "      Successfully uninstalled requests-2.28.1\n",
      "  Attempting uninstall: numpy\n",
      "    Found existing installation: numpy 1.23.2\n",
      "    Uninstalling numpy-1.23.2:\n",
      "      Successfully uninstalled numpy-1.23.2\n",
      "  Running setup.py develop for facexlib\n",
      "  Running setup.py develop for GFPGAN\n",
      "  Running setup.py develop for taming-transformers\n",
      "  Running setup.py develop for realesrgan\n",
      "  Attempting uninstall: latent-diffusion\n",
      "    Found existing installation: latent-diffusion 0.0.1\n",
      "    Uninstalling latent-diffusion-0.0.1:\n",
      "      Successfully uninstalled latent-diffusion-0.0.1\n",
      "  Running setup.py develop for latent-diffusion\n",
      "  Running setup.py develop for k-diffusion\n",
      "  Running setup.py develop for clip\n",
      "Successfully installed GFPGAN Jinja2-3.1.2 MarkupSafe-2.1.1 PyWavelets-1.3.0 absl-py-1.2.0 accelerate-0.12.0 addict-2.4.0 aiohttp-3.8.1 aiosignal-1.2.0 albumentations-0.4.3 altair-4.2.0 analytics-python-1.4.0 antlr4-python3-runtime-4.8 anyio-3.6.1 async-timeout-4.0.2 attrs-22.1.0 backoff-1.10.0 backports.zoneinfo-0.2.1 basicsr-1.4.2 bcrypt-4.0.0 blinker-1.5 cachetools-5.2.0 chardet-4.0.0 clean-fid-0.1.28 click-8.1.3 clip commonmark-0.9.1 cycler-0.11.0 decorator-5.1.1 docker-pycreds-0.4.0 einops-0.3.0 entrypoints-0.4 facexlib fastapi-0.81.0 ffmpy-0.3.0 filelock-3.8.0 filterpy-1.4.5 fonttools-4.37.1 frozenlist-1.3.1 fsspec-2022.8.0 ftfy-6.1.1 future-0.18.2 gitdb-4.0.9 gitpython-3.1.27 google-auth-2.11.0 google-auth-oauthlib-0.4.6 gradio-3.1.6 grpcio-1.47.0 h11-0.12.0 httpcore-0.15.0 httpx-0.23.0 huggingface-hub-0.9.1 idna-2.10 imageio-2.9.0 imageio-ffmpeg-0.4.2 imgaug-0.2.6 importlib-metadata-4.12.0 importlib-resources-5.9.0 jsonmerge-1.8.0 jsonschema-4.14.0 k-diffusion kiwisolver-1.4.4 kornia-0.6.0 latent-diffusion linkify-it-py-1.0.3 llvmlite-0.39.0 lmdb-1.3.0 markdown-3.4.1 markdown-it-py-2.1.0 matplotlib-3.5.3 mdit-py-plugins-0.3.0 mdurl-0.1.2 monotonic-1.6 multidict-6.0.2 networkx-2.8.6 numba-0.56.0 numpy-1.20.3 oauthlib-3.2.0 omegaconf-2.1.1 opencv-python-4.1.2.30 opencv-python-headless-4.1.2.30 orjson-3.8.0 packaging-21.3 pandas-1.4.3 paramiko-2.11.0 pathtools-0.1.2 pkgutil-resolve-name-1.3.10 promise-2.3 protobuf-3.19.4 psutil-5.9.1 pudb-2019.2 pyDeprecate-0.3.1 pyarrow-9.0.0 pyasn1-0.4.8 pyasn1-modules-0.2.8 pycryptodome-3.15.0 pydantic-1.10.0 pydeck-0.8.0b1 pydub-0.25.1 pygments-2.13.0 pympler-1.0.1 pynacl-1.5.0 pynvml-11.4.1 pyparsing-3.0.9 pyrsistent-0.18.1 python-dateutil-2.8.2 python-multipart-0.0.5 pytorch-lightning-1.4.2 pytz-2022.2.1 pytz-deprecation-shim-0.1.0.post0 pyyaml-6.0 realesrgan regex-2022.8.17 requests-2.25.1 requests-oauthlib-1.3.1 resize-right-0.0.2 rfc3986-1.5.0 rich-12.5.1 rsa-4.9 scikit-image-0.19.3 scipy-1.9.1 semver-2.13.0 sentry-sdk-1.9.5 setproctitle-1.3.2 shortuuid-1.0.9 smmap-5.0.0 sniffio-1.2.0 starlette-0.19.1 streamlit-1.12.2 taming-transformers tb-nightly-2.11.0a20220830 tensorboard-2.10.0 tensorboard-data-server-0.6.1 tensorboard-plugin-wit-1.8.1 test-tube-0.7.5 tifffile-2022.8.12 tokenizers-0.12.1 toml-0.10.2 toolz-0.12.0 torch-fidelity-0.3.0 torchdiffeq-0.2.3 torchmetrics-0.6.0 tornado-6.2 transformers-4.19.2 tzdata-2022.2 tzlocal-4.2 uc-micro-py-1.0.1 urwid-2.1.2 uvicorn-0.18.3 validators-0.20.0 wandb-0.13.2 watchdog-2.1.9 wcwidth-0.2.5 websockets-10.3 werkzeug-2.2.2 yapf-0.32.0 yarl-1.8.1 zipp-3.8.1\n",
      "\n",
      "done\n",
      "#\n",
      "# To activate this environment, use\n",
      "#\n",
      "#     $ conda activate /workspace/sd_env\n",
      "#\n",
      "# To deactivate an active environment, use\n",
      "#\n",
      "#     $ conda deactivate\n",
      "\n",
      "Retrieving notices: ...working... done\n"
     ]
    }
   ],
   "source": [
    "# Install all the dependencies for stable diffusion webui into our conda env\n",
    "!conda env update -p {conda_env} -f /workspace/stable-diffusion/environment.yaml "
   ]
  },
  {
   "cell_type": "markdown",
   "id": "bc8717dc-6e12-495f-a093-9602828bdc1f",
   "metadata": {},
   "source": [
    "### Build upscalers support\n",
    "**GFPGAN** Automatically correct distorted faces with a built-in GFPGAN option, fixes them in less than half a second  \n",
    "**ESRGAN** Boosts the resolution of images with a built-in RealESRGAN option"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "id": "9f30baec-52b6-49e3-8e8e-90e103b05075",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "/workspace/stable-diffusion/src/gfpgan\n",
      "Requirement already satisfied: basicsr in /workspace/sd_env/lib/python3.8/site-packages (1.4.2)\n",
      "Requirement already satisfied: facexlib in /workspace/stable-diffusion/src/facexlib (0.2.4)\n",
      "Requirement already satisfied: yapf in /workspace/sd_env/lib/python3.8/site-packages (0.32.0)\n",
      "Requirement already satisfied: lmdb in /workspace/sd_env/lib/python3.8/site-packages (1.3.0)\n",
      "Requirement already satisfied: opencv-python in /workspace/sd_env/lib/python3.8/site-packages (4.1.2.30)\n",
      "Requirement already satisfied: pyyaml in /workspace/sd_env/lib/python3.8/site-packages (6.0)\n",
      "Requirement already satisfied: tb-nightly in /workspace/sd_env/lib/python3.8/site-packages (2.11.0a20220830)\n",
      "/workspace/sd_env/lib/python3.8/site-packages/setuptools/installer.py:27: SetuptoolsDeprecationWarning: setuptools.installer is deprecated. Requirements should be satisfied by a PEP 517 installer.\n",
      "  warnings.warn(\n",
      "running develop\n",
      "/workspace/sd_env/lib/python3.8/site-packages/setuptools/command/easy_install.py:144: EasyInstallDeprecationWarning: easy_install command is deprecated. Use build and pip and other standards-based tools.\n",
      "  warnings.warn(\n",
      "/workspace/sd_env/lib/python3.8/site-packages/setuptools/command/install.py:34: SetuptoolsDeprecationWarning: setup.py install is deprecated. Use build and pip and other standards-based tools.\n",
      "  warnings.warn(\n",
      "running egg_info\n",
      "writing gfpgan.egg-info/PKG-INFO\n",
      "writing dependency_links to gfpgan.egg-info/dependency_links.txt\n",
      "writing requirements to gfpgan.egg-info/requires.txt\n",
      "writing top-level names to gfpgan.egg-info/top_level.txt\n",
      "reading manifest file 'gfpgan.egg-info/SOURCES.txt'\n",
      "reading manifest template 'MANIFEST.in'\n",
      "warning: no files found matching 'inputs/*'\n",
      "adding license file 'LICENSE'\n",
      "writing manifest file 'gfpgan.egg-info/SOURCES.txt'\n",
      "running build_ext\n",
      "Creating /workspace/sd_env/lib/python3.8/site-packages/gfpgan.egg-link (link to .)\n",
      "gfpgan 1.3.4 is already the active version in easy-install.pth\n",
      "\n",
      "Installed /workspace/stable-diffusion/src/gfpgan\n",
      "Processing dependencies for gfpgan==1.3.4\n",
      "Searching for wheel>=0.26\n",
      "Reading https://pypi.org/simple/wheel/\n",
      "/workspace/sd_env/lib/python3.8/site-packages/pkg_resources/__init__.py:123: PkgResourcesDeprecationWarning:  is an invalid version and will not be supported in a future release\n",
      "  warnings.warn(\n",
      "Downloading https://files.pythonhosted.org/packages/27/d6/003e593296a85fd6ed616ed962795b2f87709c3eee2bca4f6d0fe55c6d00/wheel-0.37.1-py2.py3-none-any.whl#sha256=4bdcd7d840138086126cd09254dc6195fb4fc6f01c050a1d7236f2630db1d22a\n",
      "Best match: wheel 0.37.1\n",
      "Processing wheel-0.37.1-py2.py3-none-any.whl\n",
      "Installing wheel-0.37.1-py2.py3-none-any.whl to /workspace/sd_env/lib/python3.8/site-packages\n",
      "Adding wheel 0.37.1 to easy-install.pth file\n",
      "Installing wheel script to /workspace/sd_env/bin\n",
      "\n",
      "Installed /workspace/sd_env/lib/python3.8/site-packages/wheel-0.37.1-py3.8.egg\n",
      "Searching for yapf==0.32.0\n",
      "Best match: yapf 0.32.0\n",
      "Adding yapf 0.32.0 to easy-install.pth file\n",
      "Installing yapf script to /workspace/sd_env/bin\n",
      "Installing yapf-diff script to /workspace/sd_env/bin\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for tqdm==4.64.0\n",
      "Best match: tqdm 4.64.0\n",
      "Adding tqdm 4.64.0 to easy-install.pth file\n",
      "Installing tqdm script to /workspace/sd_env/bin\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for torchvision==0.12.0\n",
      "Best match: torchvision 0.12.0\n",
      "Adding torchvision 0.12.0 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for torch==1.12.1\n",
      "Best match: torch 1.12.1\n",
      "Adding torch 1.12.1 to easy-install.pth file\n",
      "Installing convert-caffe2-to-onnx script to /workspace/sd_env/bin\n",
      "Installing convert-onnx-to-caffe2 script to /workspace/sd_env/bin\n",
      "Installing torchrun script to /workspace/sd_env/bin\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for tb-nightly==2.11.0a20220830\n",
      "Best match: tb-nightly 2.11.0a20220830\n",
      "Adding tb-nightly 2.11.0a20220830 to easy-install.pth file\n",
      "Installing tensorboard script to /workspace/sd_env/bin\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for scipy==1.9.1\n",
      "Best match: scipy 1.9.1\n",
      "Adding scipy 1.9.1 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for PyYAML==6.0\n",
      "Best match: PyYAML 6.0\n",
      "Adding PyYAML 6.0 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for opencv-python==4.1.2.30\n",
      "Best match: opencv-python 4.1.2.30\n",
      "Adding opencv-python 4.1.2.30 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for numpy==1.20.3\n",
      "Best match: numpy 1.20.3\n",
      "Adding numpy 1.20.3 to easy-install.pth file\n",
      "Installing f2py script to /workspace/sd_env/bin\n",
      "Installing f2py3 script to /workspace/sd_env/bin\n",
      "Installing f2py3.8 script to /workspace/sd_env/bin\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for lmdb==1.3.0\n",
      "Best match: lmdb 1.3.0\n",
      "Adding lmdb 1.3.0 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for facexlib==0.2.4\n",
      "Best match: facexlib 0.2.4\n",
      "facexlib 0.2.4 is already the active version in easy-install.pth\n",
      "\n",
      "Using /workspace/stable-diffusion/src/facexlib\n",
      "Searching for basicsr==1.4.2\n",
      "Best match: basicsr 1.4.2\n",
      "Adding basicsr 1.4.2 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for Pillow==9.2.0\n",
      "Best match: Pillow 9.2.0\n",
      "Adding Pillow 9.2.0 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for requests==2.25.1\n",
      "Best match: requests 2.25.1\n",
      "Adding requests 2.25.1 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for typing-extensions==4.3.0\n",
      "Best match: typing-extensions 4.3.0\n",
      "Adding typing-extensions 4.3.0 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for Werkzeug==2.2.2\n",
      "Best match: Werkzeug 2.2.2\n",
      "Adding Werkzeug 2.2.2 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for tensorboard-plugin-wit==1.8.1\n",
      "Best match: tensorboard-plugin-wit 1.8.1\n",
      "Adding tensorboard-plugin-wit 1.8.1 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for tensorboard-data-server==0.6.1\n",
      "Best match: tensorboard-data-server 0.6.1\n",
      "Adding tensorboard-data-server 0.6.1 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for setuptools==63.4.1\n",
      "Best match: setuptools 63.4.1\n",
      "Adding setuptools 63.4.1 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for protobuf==3.19.4\n",
      "Best match: protobuf 3.19.4\n",
      "Adding protobuf 3.19.4 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for Markdown==3.4.1\n",
      "Best match: Markdown 3.4.1\n",
      "Adding Markdown 3.4.1 to easy-install.pth file\n",
      "Installing markdown_py script to /workspace/sd_env/bin\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for google-auth-oauthlib==0.4.6\n",
      "Best match: google-auth-oauthlib 0.4.6\n",
      "Adding google-auth-oauthlib 0.4.6 to easy-install.pth file\n",
      "Installing google-oauthlib-tool script to /workspace/sd_env/bin\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for google-auth==2.11.0\n",
      "Best match: google-auth 2.11.0\n",
      "Adding google-auth 2.11.0 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for grpcio==1.47.0\n",
      "Best match: grpcio 1.47.0\n",
      "Adding grpcio 1.47.0 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for absl-py==1.2.0\n",
      "Best match: absl-py 1.2.0\n",
      "Adding absl-py 1.2.0 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for numba==0.56.0\n",
      "Best match: numba 0.56.0\n",
      "Adding numba 0.56.0 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for filterpy==1.4.5\n",
      "Best match: filterpy 1.4.5\n",
      "Adding filterpy 1.4.5 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for scikit-image==0.19.3\n",
      "Best match: scikit-image 0.19.3\n",
      "Adding scikit-image 0.19.3 to easy-install.pth file\n",
      "Installing skivi script to /workspace/sd_env/bin\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for future==0.18.2\n",
      "Best match: future 0.18.2\n",
      "Adding future 0.18.2 to easy-install.pth file\n",
      "Installing futurize script to /workspace/sd_env/bin\n",
      "Installing pasteurize script to /workspace/sd_env/bin\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for addict==2.4.0\n",
      "Best match: addict 2.4.0\n",
      "Adding addict 2.4.0 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for certifi==2022.6.15\n",
      "Best match: certifi 2022.6.15\n",
      "Adding certifi 2022.6.15 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for urllib3==1.26.11\n",
      "Best match: urllib3 1.26.11\n",
      "Adding urllib3 1.26.11 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for idna==2.10\n",
      "Best match: idna 2.10\n",
      "Adding idna 2.10 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for chardet==4.0.0\n",
      "Best match: chardet 4.0.0\n",
      "Adding chardet 4.0.0 to easy-install.pth file\n",
      "Installing chardetect script to /workspace/sd_env/bin\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for MarkupSafe==2.1.1\n",
      "Best match: MarkupSafe 2.1.1\n",
      "Adding MarkupSafe 2.1.1 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for importlib-metadata==4.12.0\n",
      "Best match: importlib-metadata 4.12.0\n",
      "Adding importlib-metadata 4.12.0 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for requests-oauthlib==1.3.1\n",
      "Best match: requests-oauthlib 1.3.1\n",
      "Adding requests-oauthlib 1.3.1 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for rsa==4.9\n",
      "Best match: rsa 4.9\n",
      "Adding rsa 4.9 to easy-install.pth file\n",
      "Installing pyrsa-decrypt script to /workspace/sd_env/bin\n",
      "Installing pyrsa-encrypt script to /workspace/sd_env/bin\n",
      "Installing pyrsa-keygen script to /workspace/sd_env/bin\n",
      "Installing pyrsa-priv2pub script to /workspace/sd_env/bin\n",
      "Installing pyrsa-sign script to /workspace/sd_env/bin\n",
      "Installing pyrsa-verify script to /workspace/sd_env/bin\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for six==1.16.0\n",
      "Best match: six 1.16.0\n",
      "Adding six 1.16.0 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for pyasn1-modules==0.2.8\n",
      "Best match: pyasn1-modules 0.2.8\n",
      "Adding pyasn1-modules 0.2.8 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for cachetools==5.2.0\n",
      "Best match: cachetools 5.2.0\n",
      "Adding cachetools 5.2.0 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for llvmlite==0.39.0\n",
      "Best match: llvmlite 0.39.0\n",
      "Adding llvmlite 0.39.0 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for matplotlib==3.5.3\n",
      "Best match: matplotlib 3.5.3\n",
      "Adding matplotlib 3.5.3 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for packaging==21.3\n",
      "Best match: packaging 21.3\n",
      "Adding packaging 21.3 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for PyWavelets==1.3.0\n",
      "Best match: PyWavelets 1.3.0\n",
      "Adding PyWavelets 1.3.0 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for tifffile==2022.8.12\n",
      "Best match: tifffile 2022.8.12\n",
      "Adding tifffile 2022.8.12 to easy-install.pth file\n",
      "Installing lsm2bin script to /workspace/sd_env/bin\n",
      "Installing tiff2fsspec script to /workspace/sd_env/bin\n",
      "Installing tiffcomment script to /workspace/sd_env/bin\n",
      "Installing tifffile script to /workspace/sd_env/bin\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for imageio==2.9.0\n",
      "Best match: imageio 2.9.0\n",
      "Adding imageio 2.9.0 to easy-install.pth file\n",
      "Installing imageio_download_bin script to /workspace/sd_env/bin\n",
      "Installing imageio_remove_bin script to /workspace/sd_env/bin\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for networkx==2.8.6\n",
      "Best match: networkx 2.8.6\n",
      "Adding networkx 2.8.6 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for zipp==3.8.1\n",
      "Best match: zipp 3.8.1\n",
      "Adding zipp 3.8.1 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for oauthlib==3.2.0\n",
      "Best match: oauthlib 3.2.0\n",
      "Adding oauthlib 3.2.0 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for pyasn1==0.4.8\n",
      "Best match: pyasn1 0.4.8\n",
      "Adding pyasn1 0.4.8 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for python-dateutil==2.8.2\n",
      "Best match: python-dateutil 2.8.2\n",
      "Adding python-dateutil 2.8.2 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for pyparsing==3.0.9\n",
      "Best match: pyparsing 3.0.9\n",
      "Adding pyparsing 3.0.9 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for kiwisolver==1.4.4\n",
      "Best match: kiwisolver 1.4.4\n",
      "Adding kiwisolver 1.4.4 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for fonttools==4.37.1\n",
      "Best match: fonttools 4.37.1\n",
      "Adding fonttools 4.37.1 to easy-install.pth file\n",
      "Installing fonttools script to /workspace/sd_env/bin\n",
      "Installing pyftmerge script to /workspace/sd_env/bin\n",
      "Installing pyftsubset script to /workspace/sd_env/bin\n",
      "Installing ttx script to /workspace/sd_env/bin\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Searching for cycler==0.11.0\n",
      "Best match: cycler 0.11.0\n",
      "Adding cycler 0.11.0 to easy-install.pth file\n",
      "\n",
      "Using /workspace/sd_env/lib/python3.8/site-packages\n",
      "Finished processing dependencies for gfpgan==1.3.4\n",
      "--2022-08-31 08:11:32--  https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth\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://objects.githubusercontent.com/github-production-release-asset-2e65be/349321229/e9847322-b8b1-4ec2-9620-5146eb8a9e4b?X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Credential=AKIAIWNJYAX4CSVEH53A%2F20220831%2Fus-east-1%2Fs3%2Faws4_request&X-Amz-Date=20220831T080923Z&X-Amz-Expires=300&X-Amz-Signature=a3cfe3adff7d1f689ec46ad08a19f89a397a2c62bf08088b2c0797690ddfe963&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-08-31 08:11:32--  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%2F20220831%2Fus-east-1%2Fs3%2Faws4_request&X-Amz-Date=20220831T080923Z&X-Amz-Expires=300&X-Amz-Signature=a3cfe3adff7d1f689ec46ad08a19f89a397a2c62bf08088b2c0797690ddfe963&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.109.133, 185.199.110.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.1’\n",
      "\n",
      "GFPGANv1.3.pth.1    100%[===================>] 332.48M   108MB/s    in 3.1s    \n",
      "\n",
      "2022-08-31 08:11:35 (108 MB/s) - ‘experiments/pretrained_models/GFPGANv1.3.pth.1’ saved [348632874/348632874]\n",
      "\n",
      "Requirement already satisfied: realesrgan in /workspace/stable-diffusion/src/realesrgan (0.2.5.0)\n",
      "Requirement already satisfied: basicsr>=1.3.3.11 in /workspace/sd_env/lib/python3.8/site-packages (from realesrgan) (1.4.2)\n",
      "Requirement already satisfied: facexlib>=0.2.0.3 in /workspace/stable-diffusion/src/facexlib (from realesrgan) (0.2.4)\n",
      "Requirement already satisfied: gfpgan>=0.2.1 in /workspace/stable-diffusion/src/gfpgan (from realesrgan) (1.3.4)\n",
      "Requirement already satisfied: numpy in /workspace/sd_env/lib/python3.8/site-packages (from realesrgan) (1.20.3)\n",
      "Requirement already satisfied: opencv-python in /workspace/sd_env/lib/python3.8/site-packages (from realesrgan) (4.1.2.30)\n",
      "Requirement already satisfied: Pillow in /workspace/sd_env/lib/python3.8/site-packages (from realesrgan) (9.2.0)\n",
      "Requirement already satisfied: torch>=1.7 in /workspace/sd_env/lib/python3.8/site-packages (from realesrgan) (1.12.1)\n",
      "Requirement already satisfied: torchvision in /workspace/sd_env/lib/python3.8/site-packages (from realesrgan) (0.12.0)\n",
      "Requirement already satisfied: tqdm in /workspace/sd_env/lib/python3.8/site-packages (from realesrgan) (4.64.0)\n",
      "Requirement already satisfied: tb-nightly in /workspace/sd_env/lib/python3.8/site-packages (from basicsr>=1.3.3.11->realesrgan) (2.11.0a20220830)\n",
      "Requirement already satisfied: addict in /workspace/sd_env/lib/python3.8/site-packages (from basicsr>=1.3.3.11->realesrgan) (2.4.0)\n",
      "Requirement already satisfied: scikit-image in /workspace/sd_env/lib/python3.8/site-packages (from basicsr>=1.3.3.11->realesrgan) (0.19.3)\n",
      "Requirement already satisfied: yapf in /workspace/sd_env/lib/python3.8/site-packages (from basicsr>=1.3.3.11->realesrgan) (0.32.0)\n",
      "Requirement already satisfied: lmdb in /workspace/sd_env/lib/python3.8/site-packages (from basicsr>=1.3.3.11->realesrgan) (1.3.0)\n",
      "Requirement already satisfied: future in /workspace/sd_env/lib/python3.8/site-packages (from basicsr>=1.3.3.11->realesrgan) (0.18.2)\n",
      "Requirement already satisfied: scipy in /workspace/sd_env/lib/python3.8/site-packages (from basicsr>=1.3.3.11->realesrgan) (1.9.1)\n",
      "Requirement already satisfied: requests in /workspace/sd_env/lib/python3.8/site-packages (from basicsr>=1.3.3.11->realesrgan) (2.25.1)\n",
      "Requirement already satisfied: pyyaml in /workspace/sd_env/lib/python3.8/site-packages (from basicsr>=1.3.3.11->realesrgan) (6.0)\n",
      "Requirement already satisfied: filterpy in /workspace/sd_env/lib/python3.8/site-packages (from facexlib>=0.2.0.3->realesrgan) (1.4.5)\n",
      "Requirement already satisfied: numba in /workspace/sd_env/lib/python3.8/site-packages (from facexlib>=0.2.0.3->realesrgan) (0.56.0)\n",
      "Requirement already satisfied: typing-extensions in /workspace/sd_env/lib/python3.8/site-packages (from torch>=1.7->realesrgan) (4.3.0)\n",
      "Requirement already satisfied: matplotlib in /workspace/sd_env/lib/python3.8/site-packages (from filterpy->facexlib>=0.2.0.3->realesrgan) (3.5.3)\n",
      "Requirement already satisfied: packaging>=20.0 in /workspace/sd_env/lib/python3.8/site-packages (from matplotlib->filterpy->facexlib>=0.2.0.3->realesrgan) (21.3)\n",
      "Requirement already satisfied: fonttools>=4.22.0 in /workspace/sd_env/lib/python3.8/site-packages (from matplotlib->filterpy->facexlib>=0.2.0.3->realesrgan) (4.37.1)\n",
      "Requirement already satisfied: kiwisolver>=1.0.1 in /workspace/sd_env/lib/python3.8/site-packages (from matplotlib->filterpy->facexlib>=0.2.0.3->realesrgan) (1.4.4)\n",
      "Requirement already satisfied: python-dateutil>=2.7 in /workspace/sd_env/lib/python3.8/site-packages (from matplotlib->filterpy->facexlib>=0.2.0.3->realesrgan) (2.8.2)\n",
      "Requirement already satisfied: pyparsing>=2.2.1 in /workspace/sd_env/lib/python3.8/site-packages (from matplotlib->filterpy->facexlib>=0.2.0.3->realesrgan) (3.0.9)\n",
      "Requirement already satisfied: cycler>=0.10 in /workspace/sd_env/lib/python3.8/site-packages (from matplotlib->filterpy->facexlib>=0.2.0.3->realesrgan) (0.11.0)\n",
      "Requirement already satisfied: six>=1.5 in /workspace/sd_env/lib/python3.8/site-packages (from python-dateutil>=2.7->matplotlib->filterpy->facexlib>=0.2.0.3->realesrgan) (1.16.0)\n",
      "Requirement already satisfied: importlib-metadata in /workspace/sd_env/lib/python3.8/site-packages (from numba->facexlib>=0.2.0.3->realesrgan) (4.12.0)\n",
      "Requirement already satisfied: setuptools in /workspace/sd_env/lib/python3.8/site-packages (from numba->facexlib>=0.2.0.3->realesrgan) (63.4.1)\n",
      "Requirement already satisfied: llvmlite<0.40,>=0.39.0dev0 in /workspace/sd_env/lib/python3.8/site-packages (from numba->facexlib>=0.2.0.3->realesrgan) (0.39.0)\n",
      "Requirement already satisfied: zipp>=0.5 in /workspace/sd_env/lib/python3.8/site-packages (from importlib-metadata->numba->facexlib>=0.2.0.3->realesrgan) (3.8.1)\n",
      "Requirement already satisfied: chardet<5,>=3.0.2 in /workspace/sd_env/lib/python3.8/site-packages (from requests->basicsr>=1.3.3.11->realesrgan) (4.0.0)\n",
      "Requirement already satisfied: urllib3<1.27,>=1.21.1 in /workspace/sd_env/lib/python3.8/site-packages (from requests->basicsr>=1.3.3.11->realesrgan) (1.26.11)\n",
      "Requirement already satisfied: idna<3,>=2.5 in /workspace/sd_env/lib/python3.8/site-packages (from requests->basicsr>=1.3.3.11->realesrgan) (2.10)\n",
      "Requirement already satisfied: certifi>=2017.4.17 in /workspace/sd_env/lib/python3.8/site-packages (from requests->basicsr>=1.3.3.11->realesrgan) (2022.6.15)\n",
      "Requirement already satisfied: networkx>=2.2 in /workspace/sd_env/lib/python3.8/site-packages (from scikit-image->basicsr>=1.3.3.11->realesrgan) (2.8.6)\n",
      "Requirement already satisfied: imageio>=2.4.1 in /workspace/sd_env/lib/python3.8/site-packages (from scikit-image->basicsr>=1.3.3.11->realesrgan) (2.9.0)\n",
      "Requirement already satisfied: tifffile>=2019.7.26 in /workspace/sd_env/lib/python3.8/site-packages (from scikit-image->basicsr>=1.3.3.11->realesrgan) (2022.8.12)\n",
      "Requirement already satisfied: PyWavelets>=1.1.1 in /workspace/sd_env/lib/python3.8/site-packages (from scikit-image->basicsr>=1.3.3.11->realesrgan) (1.3.0)\n",
      "Requirement already satisfied: absl-py>=0.4 in /workspace/sd_env/lib/python3.8/site-packages (from tb-nightly->basicsr>=1.3.3.11->realesrgan) (1.2.0)\n",
      "Requirement already satisfied: markdown>=2.6.8 in /workspace/sd_env/lib/python3.8/site-packages (from tb-nightly->basicsr>=1.3.3.11->realesrgan) (3.4.1)\n",
      "Requirement already satisfied: werkzeug>=1.0.1 in /workspace/sd_env/lib/python3.8/site-packages (from tb-nightly->basicsr>=1.3.3.11->realesrgan) (2.2.2)\n",
      "Requirement already satisfied: wheel>=0.26 in /workspace/sd_env/lib/python3.8/site-packages (from tb-nightly->basicsr>=1.3.3.11->realesrgan) (0.37.1)\n",
      "Requirement already satisfied: google-auth<3,>=1.6.3 in /workspace/sd_env/lib/python3.8/site-packages (from tb-nightly->basicsr>=1.3.3.11->realesrgan) (2.11.0)\n",
      "Requirement already satisfied: tensorboard-data-server<0.7.0,>=0.6.0 in /workspace/sd_env/lib/python3.8/site-packages (from tb-nightly->basicsr>=1.3.3.11->realesrgan) (0.6.1)\n",
      "Requirement already satisfied: tensorboard-plugin-wit>=1.6.0 in /workspace/sd_env/lib/python3.8/site-packages (from tb-nightly->basicsr>=1.3.3.11->realesrgan) (1.8.1)\n",
      "Requirement already satisfied: protobuf<3.20,>=3.9.2 in /workspace/sd_env/lib/python3.8/site-packages (from tb-nightly->basicsr>=1.3.3.11->realesrgan) (3.19.4)\n",
      "Requirement already satisfied: grpcio>=1.24.3 in /workspace/sd_env/lib/python3.8/site-packages (from tb-nightly->basicsr>=1.3.3.11->realesrgan) (1.47.0)\n",
      "Requirement already satisfied: google-auth-oauthlib<0.5,>=0.4.1 in /workspace/sd_env/lib/python3.8/site-packages (from tb-nightly->basicsr>=1.3.3.11->realesrgan) (0.4.6)\n",
      "Requirement already satisfied: pyasn1-modules>=0.2.1 in /workspace/sd_env/lib/python3.8/site-packages (from google-auth<3,>=1.6.3->tb-nightly->basicsr>=1.3.3.11->realesrgan) (0.2.8)\n",
      "Requirement already satisfied: rsa<5,>=3.1.4 in /workspace/sd_env/lib/python3.8/site-packages (from google-auth<3,>=1.6.3->tb-nightly->basicsr>=1.3.3.11->realesrgan) (4.9)\n",
      "Requirement already satisfied: cachetools<6.0,>=2.0.0 in /workspace/sd_env/lib/python3.8/site-packages (from google-auth<3,>=1.6.3->tb-nightly->basicsr>=1.3.3.11->realesrgan) (5.2.0)\n",
      "Requirement already satisfied: requests-oauthlib>=0.7.0 in /workspace/sd_env/lib/python3.8/site-packages (from google-auth-oauthlib<0.5,>=0.4.1->tb-nightly->basicsr>=1.3.3.11->realesrgan) (1.3.1)\n",
      "Requirement already satisfied: pyasn1<0.5.0,>=0.4.6 in /workspace/sd_env/lib/python3.8/site-packages (from pyasn1-modules>=0.2.1->google-auth<3,>=1.6.3->tb-nightly->basicsr>=1.3.3.11->realesrgan) (0.4.8)\n",
      "Requirement already satisfied: oauthlib>=3.0.0 in /workspace/sd_env/lib/python3.8/site-packages (from requests-oauthlib>=0.7.0->google-auth-oauthlib<0.5,>=0.4.1->tb-nightly->basicsr>=1.3.3.11->realesrgan) (3.2.0)\n",
      "Requirement already satisfied: MarkupSafe>=2.1.1 in /workspace/sd_env/lib/python3.8/site-packages (from werkzeug>=1.0.1->tb-nightly->basicsr>=1.3.3.11->realesrgan) (2.1.1)\n",
      "/workspace/stable-diffusion/src/realesrgan\n",
      "--2022-08-31 08:11:38--  https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth\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://objects.githubusercontent.com/github-production-release-asset-2e65be/387326890/08f0e941-ebb7-48f0-9d6a-73e87b710e7e?X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Credential=AKIAIWNJYAX4CSVEH53A%2F20220831%2Fus-east-1%2Fs3%2Faws4_request&X-Amz-Date=20220831T081005Z&X-Amz-Expires=300&X-Amz-Signature=ae7a8dfe69f30177b95dd0eb191856fe1434b32dc42d4d9c7ac70c5ad6e8c7c6&X-Amz-SignedHeaders=host&actor_id=0&key_id=0&repo_id=387326890&response-content-disposition=attachment%3B%20filename%3DRealESRGAN_x4plus.pth&response-content-type=application%2Foctet-stream [following]\n",
      "--2022-08-31 08:11:38--  https://objects.githubusercontent.com/github-production-release-asset-2e65be/387326890/08f0e941-ebb7-48f0-9d6a-73e87b710e7e?X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Credential=AKIAIWNJYAX4CSVEH53A%2F20220831%2Fus-east-1%2Fs3%2Faws4_request&X-Amz-Date=20220831T081005Z&X-Amz-Expires=300&X-Amz-Signature=ae7a8dfe69f30177b95dd0eb191856fe1434b32dc42d4d9c7ac70c5ad6e8c7c6&X-Amz-SignedHeaders=host&actor_id=0&key_id=0&repo_id=387326890&response-content-disposition=attachment%3B%20filename%3DRealESRGAN_x4plus.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.109.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: 67040989 (64M) [application/octet-stream]\n",
      "Saving to: ‘experiments/pretrained_models/RealESRGAN_x4plus.pth.1’\n",
      "\n",
      "RealESRGAN_x4plus.p 100%[===================>]  63.93M  29.7MB/s    in 2.2s    \n",
      "\n",
      "2022-08-31 08:11:40 (29.7 MB/s) - ‘experiments/pretrained_models/RealESRGAN_x4plus.pth.1’ saved [67040989/67040989]\n",
      "\n",
      "--2022-08-31 08:11:41--  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.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://objects.githubusercontent.com/github-production-release-asset-2e65be/387326890/4f59d7c8-d03f-494e-8595-ae23af075393?X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Credential=AKIAIWNJYAX4CSVEH53A%2F20220831%2Fus-east-1%2Fs3%2Faws4_request&X-Amz-Date=20220831T081008Z&X-Amz-Expires=300&X-Amz-Signature=acce79ad242e3e44f2407b84582aef2b7dcffb080f8b07f98686aa38c0e47f41&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-08-31 08:11:41--  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%2F20220831%2Fus-east-1%2Fs3%2Faws4_request&X-Amz-Date=20220831T081008Z&X-Amz-Expires=300&X-Amz-Signature=acce79ad242e3e44f2407b84582aef2b7dcffb080f8b07f98686aa38c0e47f41&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.109.133, 185.199.110.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.1’\n",
      "\n",
      "RealESRGAN_x4plus_a 100%[===================>]  17.11M  89.6MB/s    in 0.2s    \n",
      "\n",
      "2022-08-31 08:11:41 (89.6 MB/s) - ‘experiments/pretrained_models/RealESRGAN_x4plus_anime_6B.pth.1’ saved [17938799/17938799]\n",
      "\n",
      "/workspace/stable-diffusion\n",
      "--2022-08-31 08:11:42--  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-08-31 08:11:42--  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-08-31 08:11:42--  https://raw.githubusercontent.com/matomo-org/travis-scripts/master/fonts/Arial.ttf\n",
      "Resolving raw.githubusercontent.com (raw.githubusercontent.com)... 185.199.108.133, 185.199.109.133, 185.199.110.133, ...\n",
      "Connecting to raw.githubusercontent.com (raw.githubusercontent.com)|185.199.108.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.03s   \n",
      "\n",
      "2022-08-31 08:11:42 (9.22 MB/s) - ‘arial.ttf’ saved [275572/275572]\n",
      "\n",
      "/workspace\n"
     ]
    }
   ],
   "source": [
    "if add_GFP:\n",
    "    %cd {wd}/stable-diffusion/src/gfpgan/\n",
    "    !{pip_path} install basicsr facexlib yapf lmdb opencv-python pyyaml tb-nightly --no-deps\n",
    "    !{python_path} setup.py develop\n",
    "    !wget https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth -P experiments/pretrained_models\n",
    "if add_ESR:\n",
    "    !{pip_path} install realesrgan\n",
    "    %cd {wd}/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",
    "\n",
    "%cd {wd}/stable-diffusion/\n",
    "!wget https://github.com/matomo-org/travis-scripts/blob/master/fonts/Arial.ttf?raw=true -O arial.ttf\n",
    "%cd {wd}"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "id": "0404d050-445c-40c8-ba7f-5e0924df2eeb",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Collecting package metadata (current_repodata.json): done\n",
      "Solving environment: done\n",
      "\n",
      "## Package Plan ##\n",
      "\n",
      "  environment location: /workspace/sd_env\n",
      "\n",
      "  added / updated specs:\n",
      "    - glib\n",
      "\n",
      "\n",
      "The following packages will be downloaded:\n",
      "\n",
      "    package                    |            build\n",
      "    ---------------------------|-----------------\n",
      "    glib-2.69.1                |       h4ff587b_1         1.7 MB\n",
      "    pcre-8.45                  |       h295c915_0         207 KB\n",
      "    ------------------------------------------------------------\n",
      "                                           Total:         1.9 MB\n",
      "\n",
      "The following NEW packages will be INSTALLED:\n",
      "\n",
      "  glib               pkgs/main/linux-64::glib-2.69.1-h4ff587b_1\n",
      "  pcre               pkgs/main/linux-64::pcre-8.45-h295c915_0\n",
      "\n",
      "\n",
      "\n",
      "Downloading and Extracting Packages\n",
      "glib-2.69.1          | 1.7 MB    | ##################################### | 100% \n",
      "pcre-8.45            | 207 KB    | ##################################### | 100% \n",
      "Preparing transaction: done\n",
      "Verifying transaction: done\n",
      "Executing transaction: done\n",
      "Retrieving notices: ...working... done\n"
     ]
    }
   ],
   "source": [
    "# Install glib, which is necessary to run esrgan/gfpgan\n",
    "!conda install -p {conda_env} glib -y"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "85167150-2c1c-411a-871f-300a3b215134",
   "metadata": {},
   "source": [
    "### Fetch Model Weights for stablediffusion"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "id": "fd5238ca-1d40-4ced-bb79-2cd4aa595c54",
   "metadata": {},
   "outputs": [],
   "source": [
    "import os\n",
    "os.makedirs(models_path, exist_ok=True)\n",
    "os.makedirs(output_path, exist_ok=True)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "id": "c8f79037-e9ca-4b6f-bdd7-fd3866bd90bc",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "  % Total    % Received % Xferd  Average Speed   Time    Time     Time  Current\n",
      "                                 Dload  Upload   Total   Spent    Left  Speed\n",
      "100 4067M  100 4067M    0     0  67.6M      0  0:01:00  0:01:00 --:--:-- 64.9M\n"
     ]
    }
   ],
   "source": [
    "# Snag the weights from a publicly hosted copy. If this goes down, use huggingface as a backup.\n",
    "!curl https://www.googleapis.com/storage/v1/b/aai-blog-files/o/sd-v1-4.ckpt?alt=media > {wd}/models/sd-v1-4.ckpt"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "47b02b6a",
   "metadata": {},
   "source": [
    "***"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "22a1a16a-98bf-4cae-8ead-ce68459e3300",
   "metadata": {},
   "source": [
    "## 3. Run the Stable Diffusion web UI"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "3c8a30b4-10b8-4115-839c-517073c70787",
   "metadata": {},
   "source": [
    "### Launch The WebUI!"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "8ceb99af",
   "metadata": {},
   "source": [
    "This cell will launch the actual web app!\n",
    "\n",
    "If you encounter errors or exceptions here, please make sure all the cells above have been run, in order, with no errors. Remember that even if you've already set this up, the cells in Section 1 must be rerun every time your environment is recreated.\n",
    "\n",
    "*The web-app takes between 1-2 minutes to fully start up.*"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "id": "7a9e4e1f-8930-4540-ac04-9131eb3cec8c",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "/workspace/stable-diffusion\n",
      "Downloading: \"https://github.com/xinntao/facexlib/releases/download/v0.1.0/detection_Resnet50_Final.pth\" to /workspace/stable-diffusion/src/facexlib/facexlib/weights/detection_Resnet50_Final.pth\n",
      "\n",
      "100%|████████████████████████████████████████| 104M/104M [00:03<00:00, 27.5MB/s]\n",
      "Downloading: \"https://github.com/xinntao/facexlib/releases/download/v0.2.2/parsing_parsenet.pth\" to /workspace/stable-diffusion/src/facexlib/facexlib/weights/parsing_parsenet.pth\n",
      "\n",
      "100%|███████████████████████████████████████| 81.4M/81.4M [00:00<00:00, 111MB/s]\n",
      "Loaded GFPGAN\n",
      "Loaded RealESRGAN with model RealESRGAN_x4plus\n",
      "Loading model from /workspace/models/sd-v1-4.ckpt\n",
      "Global Step: 470000\n",
      "UNet: Running in eps-prediction mode\n",
      "^C\n",
      "Traceback (most recent call last):\n",
      "  File \"/workspace/stable-diffusion/scripts/webui.py\", line 359, in <module>\n",
      "    _, _ = model.load_state_dict(sd, strict=False)\n",
      "  File \"/workspace/sd_env/lib/python3.8/site-packages/torch/nn/modules/module.py\", line 1483, in load_state_dict\n",
      "    load(self)\n",
      "  File \"/workspace/sd_env/lib/python3.8/site-packages/torch/nn/modules/module.py\", line 1481, in load\n",
      "    load(child, prefix + name + '.')\n",
      "  File \"/workspace/sd_env/lib/python3.8/site-packages/torch/nn/modules/module.py\", line 1481, in load\n",
      "    load(child, prefix + name + '.')\n",
      "  File \"/workspace/sd_env/lib/python3.8/site-packages/torch/nn/modules/module.py\", line 1481, in load\n",
      "    load(child, prefix + name + '.')\n",
      "  [Previous line repeated 6 more times]\n",
      "  File \"/workspace/sd_env/lib/python3.8/site-packages/torch/nn/modules/module.py\", line 1477, in load\n",
      "    module._load_from_state_dict(\n",
      "  File \"/workspace/sd_env/lib/python3.8/site-packages/torch/nn/modules/module.py\", line 1413, in _load_from_state_dict\n",
      "    param.copy_(input_param)\n",
      "KeyboardInterrupt\n"
     ]
    }
   ],
   "source": [
    "\n",
    "#@markdown # Important - click the public URL to launch WebUI in another tab\n",
    "\n",
    "%cd {wd}/stable-diffusion\n",
    "!{python_path} {wd}/stable-diffusion/scripts/webui.py \\\n",
    "--ckpt '{models_path}/sd-v1-4.ckpt' \\\n",
    "--outdir '{output_path}' \\\n",
    "--share {vars}"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "029fdd4f-1a26-47b1-9a42-19eb1a2993a2",
   "metadata": {},
   "source": [
    "^ I interrupted the code above before it finished, but when complete it will give you a public Gradio URL you can use to access your webapp. Have fun!"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "1131b65d-d4ea-49c0-bd98-691c55b56a17",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "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.7.13"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}