multimodal/stable_diffusion/webui_kaggle.ipynb

{
 "metadata": {
  "kernelspec": {
   "language": "python",
   "display_name": "Python 3",
   "name": "python3"
  },
  "language_info": {
   "name": "python",
   "version": "3.10.10",
   "mimetype": "text/x-python",
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "pygments_lexer": "ipython3",
   "nbconvert_exporter": "python",
   "file_extension": ".py"
  }
 },
 "nbformat_minor": 4,
 "nbformat": 4,
 "cells": [
  {
   "cell_type": "code",
   "source": [
    "# 拷贝代码\n",
    "!git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui"
   ],
   "metadata": {
    "execution": {
     "iopub.status.busy": "2023-06-07T13:41:17.738247Z",
     "iopub.execute_input": "2023-06-07T13:41:17.738737Z",
     "iopub.status.idle": "2023-06-07T13:41:22.458053Z",
     "shell.execute_reply.started": "2023-06-07T13:41:17.738696Z",
     "shell.execute_reply": "2023-06-07T13:41:22.456911Z"
    },
    "trusted": true
   },
   "execution_count": 1,
   "outputs": [
    {
     "name": "stdout",
     "text": "Cloning into 'stable-diffusion-webui'...\nremote: Enumerating objects: 22291, done.\u001B[K\nremote: Counting objects: 100% (437/437), done.\u001B[K\nremote: Compressing objects: 100% (259/259), done.\u001B[K\nremote: Total 22291 (delta 243), reused 296 (delta 158), pack-reused 21854\u001B[K\nReceiving objects: 100% (22291/22291), 30.32 MiB | 18.27 MiB/s, done.\nResolving deltas: 100% (15574/15574), done.\n",
     "output_type": "stream"
    }
   ]
  },
  {
   "cell_type": "code",
   "source": [
    "#从c站下载模型，放到/kaggle/working/目录中\n",
    "%cd /kaggle/working/stable-diffusion-webui/models\n",
    "# SD的Base模型\n",
    "!wget -O DreamShaper.safetensors https://civitai.com/api/download/models/78164\n",
    "\n",
    "# Lora权重\n",
    "!mkdir Lora\n",
    "!wget -O Lora/nijiMecha.safetensors https://civitai.com/api/download/models/70067\n",
    "\n",
    "!ls\n",
    "!ls Lora/\n",
    "\n",
    "############ 这里也可以把常用的模型放在自己的dataset中，这样每次启动的时候就可以直接把模型加进来了(Add Data) ############"
   ],
   "metadata": {
    "execution": {
     "iopub.status.busy": "2023-06-07T13:41:24.827330Z",
     "iopub.execute_input": "2023-06-07T13:41:24.827770Z",
     "iopub.status.idle": "2023-06-07T13:42:11.615086Z",
     "shell.execute_reply.started": "2023-06-07T13:41:24.827735Z",
     "shell.execute_reply": "2023-06-07T13:42:11.613902Z"
    },
    "trusted": true
   },
   "execution_count": 2,
   "outputs": [
    {
     "name": "stdout",
     "text": "/kaggle/working/stable-diffusion-webui/models\n--2023-06-07 13:41:25--  https://civitai.com/api/download/models/78164\nResolving civitai.com (civitai.com)... 104.18.22.206, 104.18.23.206, 2606:4700::6812:17ce, ...\nConnecting to civitai.com (civitai.com)|104.18.22.206|:443... connected.\nHTTP request sent, awaiting response... 307 Temporary Redirect\nLocation: https://civitai-delivery-worker-prod-2023-06-01.5ac0637cfd0766c97916cefa3764fbdf.r2.cloudflarestorage.com/53515/model/dreamshaper6Bakedvae.2vKL.safetensors?X-Amz-Expires=86400&response-content-disposition=attachment%3B%20filename%3D%22dreamshaper_6BakedVae.safetensors%22&X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Credential=2fea663d76bd24a496545da373d610fc/20230607/us-east-1/s3/aws4_request&X-Amz-Date=20230607T134126Z&X-Amz-SignedHeaders=host&X-Amz-Signature=d558ba3404a6651a919937216ca0d0a521353165687a987e9dae12a85477ab81 [following]\n--2023-06-07 13:41:26--  https://civitai-delivery-worker-prod-2023-06-01.5ac0637cfd0766c97916cefa3764fbdf.r2.cloudflarestorage.com/53515/model/dreamshaper6Bakedvae.2vKL.safetensors?X-Amz-Expires=86400&response-content-disposition=attachment%3B%20filename%3D%22dreamshaper_6BakedVae.safetensors%22&X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Credential=2fea663d76bd24a496545da373d610fc/20230607/us-east-1/s3/aws4_request&X-Amz-Date=20230607T134126Z&X-Amz-SignedHeaders=host&X-Amz-Signature=d558ba3404a6651a919937216ca0d0a521353165687a987e9dae12a85477ab81\nResolving civitai-delivery-worker-prod-2023-06-01.5ac0637cfd0766c97916cefa3764fbdf.r2.cloudflarestorage.com (civitai-delivery-worker-prod-2023-06-01.5ac0637cfd0766c97916cefa3764fbdf.r2.cloudflarestorage.com)... 104.18.9.90, 104.18.8.90, 2606:4700::6812:85a, ...\nConnecting to civitai-delivery-worker-prod-2023-06-01.5ac0637cfd0766c97916cefa3764fbdf.r2.cloudflarestorage.com (civitai-delivery-worker-prod-2023-06-01.5ac0637cfd0766c97916cefa3764fbdf.r2.cloudflarestorage.com)|104.18.9.90|:443... connected.\nHTTP request sent, awaiting response... 200 OK\nLength: 2132625431 (2.0G) [application/octet-stream]\nSaving to: ‘DreamShaper.safetensors’\n\nDreamShaper.safeten 100%[===================>]   1.99G  72.3MB/s    in 34s     \n\n2023-06-07 13:42:00 (60.2 MB/s) - ‘DreamShaper.safetensors’ saved [2132625431/2132625431]\n\n--2023-06-07 13:42:02--  https://civitai.com/api/download/models/70067\nResolving civitai.com (civitai.com)... 104.18.22.206, 104.18.23.206, 2606:4700::6812:17ce, ...\nConnecting to civitai.com (civitai.com)|104.18.22.206|:443... connected.\nHTTP request sent, awaiting response... 307 Temporary Redirect\nLocation: https://civitai-delivery-worker-prod-2023-06-01.5ac0637cfd0766c97916cefa3764fbdf.r2.cloudflarestorage.com/137281/model/nijimecha.1N3H.safetensors?X-Amz-Expires=86400&response-content-disposition=attachment%3B%20filename%3D%22nijiMecha.safetensors%22&X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Credential=2fea663d76bd24a496545da373d610fc/20230607/us-east-1/s3/aws4_request&X-Amz-Date=20230607T134203Z&X-Amz-SignedHeaders=host&X-Amz-Signature=bb4a90c6b2b34bf763c88919895a81f8803ed71079d76a5d7405c14703ea169f [following]\n--2023-06-07 13:42:03--  https://civitai-delivery-worker-prod-2023-06-01.5ac0637cfd0766c97916cefa3764fbdf.r2.cloudflarestorage.com/137281/model/nijimecha.1N3H.safetensors?X-Amz-Expires=86400&response-content-disposition=attachment%3B%20filename%3D%22nijiMecha.safetensors%22&X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Credential=2fea663d76bd24a496545da373d610fc/20230607/us-east-1/s3/aws4_request&X-Amz-Date=20230607T134203Z&X-Amz-SignedHeaders=host&X-Amz-Signature=bb4a90c6b2b34bf763c88919895a81f8803ed71079d76a5d7405c14703ea169f\nResolving civitai-delivery-worker-prod-2023-06-01.5ac0637cfd0766c97916cefa3764fbdf.r2.cloudflarestorage.com (civitai-delivery-worker-prod-2023-06-01.5ac0637cfd0766c97916cefa3764fbdf.r2.cloudflarestorage.com)... 104.18.9.90, 104.18.8.90, 2606:4700::6812:85a, ...\nConnecting to civitai-delivery-worker-prod-2023-06-01.5ac0637cfd0766c97916cefa3764fbdf.r2.cloudflarestorage.com (civitai-delivery-worker-prod-2023-06-01.5ac0637cfd0766c97916cefa3764fbdf.r2.cloudflarestorage.com)|104.18.9.90|:443... connected.\nHTTP request sent, awaiting response... 200 OK\nLength: 151115393 (144M) [application/octet-stream]\nSaving to: ‘Lora/nijiMecha.safetensors’\n\nLora/nijiMecha.safe 100%[===================>] 144.11M  29.6MB/s    in 6.1s    \n\n2023-06-07 13:42:09 (23.6 MB/s) - ‘Lora/nijiMecha.safetensors’ saved [151115393/151115393]\n\nDreamShaper.safetensors  Stable-diffusion  VAE-approx  karlo\nLora\t\t\t VAE\t\t   deepbooru\nnijiMecha.safetensors\n",
     "output_type": "stream"
    }
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "outputs": [],
   "source": [
    "# 由于kaggle升级了Python的版本到3.10.10，但3.10.6以上的torch支持有问题，因此需要新建一个3.10.6的Python环境\n",
    "!conda config --remove-key channels\n",
    "!conda config --append channels conda-forge --append channels defaults\n",
    "!conda create -n sd python=3.10.6 -y -c conda-forge"
   ],
   "metadata": {
    "collapsed": false
   }
  },
  {
   "cell_type": "code",
   "source": [
    "#执行启动UI\n",
    "%cd /kaggle/working/stable-diffusion-webui\n",
    "!/opt/conda/envs/sd/bin/python launch.py --no-half --share --enable-insecure-extension-access\n",
    "# !python launch.py --no-half --share --enable-insecure-extension-access\n",
    "# 中断后再启动，可不用重新安装环境\n",
    "# !python launch.py --no-half --share --enable-insecure-extension-access --skip-prepare-environment"
   ],
   "metadata": {
    "execution": {
     "iopub.status.busy": "2023-06-07T13:42:48.988418Z",
     "iopub.execute_input": "2023-06-07T13:42:48.989431Z"
    },
    "trusted": true
   },
   "execution_count": null,
   "outputs": [
    {
     "name": "stdout",
     "text": "/kaggle/working/stable-diffusion-webui\nPython 3.10.10 | packaged by conda-forge | (main, Mar 24 2023, 20:08:06) [GCC 11.3.0]\nVersion: v1.3.2\nCommit hash: baf6946e06249c5af9851c60171692c44ef633e0\nInstalling gfpgan\nInstalling clip\nInstalling open_clip\nCloning Stable Diffusion into /kaggle/working/stable-diffusion-webui/repositories/stable-diffusion-stability-ai...\nCloning Taming Transformers into /kaggle/working/stable-diffusion-webui/repositories/taming-transformers...\nCloning K-diffusion into /kaggle/working/stable-diffusion-webui/repositories/k-diffusion...\nCloning CodeFormer into /kaggle/working/stable-diffusion-webui/repositories/CodeFormer...\nCloning BLIP into /kaggle/working/stable-diffusion-webui/repositories/BLIP...\nInstalling requirements for CodeFormer\nInstalling requirements\nLaunching Web UI with arguments: --no-half --share --enable-insecure-extension-access\n/opt/conda/lib/python3.10/site-packages/scipy/__init__.py:146: UserWarning: A NumPy version >=1.16.5 and <1.23.0 is required for this version of SciPy (detected version 1.23.5\n  warnings.warn(f\"A NumPy version >={np_minversion} and <{np_maxversion}\"\nNo module 'xformers'. Proceeding without it.\nMoving DreamShaper.safetensors from /kaggle/working/stable-diffusion-webui/models to /kaggle/working/stable-diffusion-webui/models/Stable-diffusion.\nCalculating sha256 for /kaggle/working/stable-diffusion-webui/models/Stable-diffusion/DreamShaper.safetensors: Running on local URL:  http://127.0.0.1:7860\nc249d7853b061e77fbe96ddb20687714591b518d30de3222ed38714a4e48e866\nLoading weights [c249d7853b] from /kaggle/working/stable-diffusion-webui/models/Stable-diffusion/DreamShaper.safetensors\nCreating model from config: /kaggle/working/stable-diffusion-webui/configs/v1-inference.yaml\nLatentDiffusion: Running in eps-prediction mode\nDiffusionWrapper has 859.52 M params.\nDownloading (…)olve/main/vocab.json: 100%|███| 961k/961k [00:00<00:00, 1.21MB/s]\nDownloading (…)olve/main/merges.txt:   0%|           | 0.00/525k [00:00<?, ?B/s]Running on public URL: https://66e3320508feb4575b.gradio.live\n\nThis share link expires in 72 hours. For free permanent hosting and GPU upgrades (NEW!), check out Spaces: https://huggingface.co/spaces\nStartup time: 29.6s (import torch: 8.9s, import gradio: 0.9s, import ldm: 0.5s, other imports: 6.5s, setup codeformer: 0.2s, load scripts: 1.5s, create ui: 0.5s, gradio launch: 10.5s).\nDownloading (…)olve/main/merges.txt: 100%|████| 525k/525k [00:00<00:00, 883kB/s]\nDownloading (…)cial_tokens_map.json: 100%|██████| 389/389 [00:00<00:00, 154kB/s]\nDownloading (…)okenizer_config.json: 100%|██████| 905/905 [00:00<00:00, 635kB/s]\nDownloading (…)lve/main/config.json: 100%|█| 4.52k/4.52k [00:00<00:00, 3.03MB/s]\nApplying optimization: Doggettx... done.\nTextual inversion embeddings loaded(0): \nModel loaded in 19.0s (calculate hash: 6.6s, load weights from disk: 0.4s, create model: 6.5s, apply weights to model: 4.3s, move model to device: 1.2s).\n  0%|                                                    | 0/25 [00:00<?, ?it/s]\n  4%|█▊                                          | 1/25 [00:09<03:46,  9.45s/it]\u001B[A\n  8%|███▌                                        | 2/25 [00:12<02:14,  5.84s/it]\u001B[A\n 12%|█████▎                                      | 3/25 [00:16<01:43,  4.69s/it]\u001B[A\n 16%|███████                                     | 4/25 [00:19<01:26,  4.14s/it]\u001B[A\n 20%|████████▊                                   | 5/25 [00:22<01:16,  3.84s/it]\u001B[A\n 24%|██████████▌                                 | 6/25 [00:26<01:09,  3.66s/it]\u001B[A\n 28%|████████████▎                               | 7/25 [00:29<01:03,  3.55s/it]\u001B[A\n 32%|██████████████                              | 8/25 [00:32<00:59,  3.48s/it]\u001B[A\n 36%|███████████████▊                            | 9/25 [00:35<00:54,  3.42s/it]\u001B[A\n 40%|█████████████████▏                         | 10/25 [00:39<00:50,  3.39s/it]\u001B[A\n 44%|██████████████████▉                        | 11/25 [00:42<00:47,  3.36s/it]\u001B[A\n 48%|████████████████████▋                      | 12/25 [00:45<00:43,  3.35s/it]\u001B[A\n 52%|██████████████████████▎                    | 13/25 [00:49<00:40,  3.34s/it]\u001B[A\n 56%|████████████████████████                   | 14/25 [00:52<00:36,  3.33s/it]\u001B[A\n 60%|█████████████████████████▊                 | 15/25 [00:55<00:33,  3.32s/it]\u001B[A\n 64%|███████████████████████████▌               | 16/25 [00:59<00:29,  3.32s/it]\u001B[A\n 68%|█████████████████████████████▏             | 17/25 [01:02<00:26,  3.32s/it]\u001B[A\n 72%|██████████████████████████████▉            | 18/25 [01:05<00:23,  3.32s/it]\u001B[A\n 76%|████████████████████████████████▋          | 19/25 [01:09<00:19,  3.31s/it]\u001B[A\n 80%|██████████████████████████████████▍        | 20/25 [01:12<00:16,  3.31s/it]\u001B[A\n 84%|████████████████████████████████████       | 21/25 [01:15<00:13,  3.31s/it]\u001B[A\n 88%|█████████████████████████████████████▊     | 22/25 [01:18<00:09,  3.31s/it]\u001B[A\n 92%|███████████████████████████████████████▌   | 23/25 [01:22<00:06,  3.31s/it]\u001B[A\n 96%|█████████████████████████████████████████▎ | 24/25 [01:25<00:03,  3.31s/it]\u001B[A\n100%|███████████████████████████████████████████| 25/25 [01:28<00:00,  3.56s/it]\u001B[A\n\nTotal progress: 100%|███████████████████████████| 25/25 [01:21<00:00,  3.26s/it]\u001B[A\n",
     "output_type": "stream"
    }
   ]
  },
  {
   "cell_type": "code",
   "source": [
    "# 设置--share，日志会打印出公网访问的链接\n",
    "# 到此，你就可以在浏览器访问你的webui了\n",
    "# Running on public URL: https://66e3320508feb4575b.gradio.live"
   ],
   "metadata": {},
   "execution_count": null,
   "outputs": []
  }
 ]
}