Skip to content

Commit

Permalink
Merge remote-tracking branch 'sa-shiro/main'
Browse files Browse the repository at this point in the history
  • Loading branch information
@ddPn08
ddPn08 committed Jan 22, 2023
2 parents 452f543 + dba998d commit fe5812f
Showing 1 changed file with 52 additions and 25 deletions.
77 changes: 52 additions & 25 deletions automatic1111.ipynb
View file
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,8 @@
{
"cell_type": "markdown",
"metadata": {
"id": "view-in-github",
"colab_type": "text"
"colab_type": "text",
"id": "view-in-github"
},
"source": [
"<a href=\"https://colab.research.google.com/github/ddPn08/automatic1111-colab/blob/main/automatic1111.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
Expand Down Expand Up @@ -96,8 +96,8 @@
"cell_type": "code",
"execution_count": null,
"metadata": {
"id": "Mls4_48XOrTd",
"cellView": "form"
"cellView": "form",
"id": "Mls4_48XOrTd"
},
"outputs": [],
"source": [
Expand All @@ -107,20 +107,30 @@
"\n",
"%cd /content/\n",
"\n",
"data_dir = \"/content/data\"\n",
"data_dir = \"/content/data\" # @param {type:\"string\"}\n",
"\n",
"# @markdown **Optional | Download the model if it isn't already in the `{data_dir}/models` folder**\n",
"\n",
"# @markdown Get huggingface access token from [here](https://huggingface.co/settings/tokens)\n",
"auth_token = \"\" # @param {type:\"string\"}\n",
"\n",
"download_if_missing = False # @param {type:\"boolean\"}\n",
"model_url = \"https://huggingface.co/hakurei/waifu-diffusion-v1-3/resolve/main/wd-v1-3-float16.ckpt\" # @param {type:\"string\"}\n",
"config_url = \"\" # @param {type:\"string\"}\n",
"# @markdown If you use the SD 2.1 model, select its config from the dropdown\n",
"model = \"https://huggingface.co/Linaqruf/anything-v3.0/resolve/main/Anything-V3.0-pruned.ckpt\" #@param [\"https://huggingface.co/hakurei/waifu-diffusion-v1-3/resolve/main/wd-v1-3-float16.ckpt\", \"https://huggingface.co/Linaqruf/anything-v3.0/resolve/main/Anything-V3.0-pruned.ckpt\", \"https://huggingface.co/CompVis/stable-diffusion-v-1-4-original/resolve/main/sd-v1-4.ckpt\", \"https://huggingface.co/stabilityai/stable-diffusion-2-1-base/resolve/main/v2-1_512-ema-pruned.ckpt\"] {allow-input: true}\n",
"model_filename = \"\" # @param {type:\"string\"}\n",
"config_url = \"\" #@param [\"\", \"https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inference.yaml\"] {allow-input: true}\n",
"config_filename = \"\" # @param {type:\"string\"}\n",
"auth_token = \"\" # @param {type:\"string\"}\n",
"use_vae = True # @param {type:\"boolean\"}\n",
"vae = \"https://huggingface.co/Linaqruf/anything-v3.0/resolve/main/Anything-V3.0.vae.pt\" #@param [\"https://huggingface.co/Linaqruf/anything-v3.0/resolve/main/Anything-V3.0.vae.pt\"] {allow-input: true}\n",
"vae_filename = \"\" # @param {type:\"string\"}\n",
"\n",
"model_url = model\n",
"vae_url = vae\n",
"\n",
"\n",
"# @markdown **Optional | Use Google Drive**\n",
"mount_google_drive = True # @param {type:\"boolean\"}\n",
"force_model_download_locally = False # @param {type:\"boolean\"}\n",
"force_remount = False # @param {type:\"boolean\"}\n",
"\n",
"import os\n",
Expand Down Expand Up @@ -150,11 +160,15 @@
"scripts_path = f\"{data_dir}/scripts\"\n",
"extensions_file_path = f\"{data_dir}/extensions.txt\"\n",
"\n",
"if force_model_download_locally:\n",
" models_path = \"/content/models\"\n",
"\n",
"os.makedirs(models_path, exist_ok=True)\n",
"os.makedirs(output_path, exist_ok=True)\n",
"os.makedirs(config_path, exist_ok=True)\n",
"os.makedirs(scripts_path, exist_ok=True)\n",
"os.makedirs(f\"{models_path}/Stable-diffusion\", exist_ok=True)\n",
"os.makedirs(f\"{models_path}/VAE\", exist_ok=True)\n",
"\n",
"for script in os.listdir(scripts_path):\n",
" ! rm -Rf stable-diffusion-webui/scripts/{script} && ln -s {scripts_path}/{script} stable-diffusion-webui/scripts/{script}\n",
Expand Down Expand Up @@ -182,7 +196,11 @@
" for [url, filename] in [[model_url, model_filename], [config_url, config_filename]]:\n",
" filename = filename or url.split(\"/\")[-1]\n",
" if not os.path.exists(f\"{models_path}/Stable-diffusion/{filename}\"):\n",
" ! curl -LJ {url} -o {models_path}/Stable-diffusion/{filename} {'-H \"Authorization: Bearer ' + auth_token + '\"' if auth_token else \"\"}"
" ! curl -LJ {url} -o {models_path}/Stable-diffusion/{filename} {'-H \"Authorization: Bearer ' + auth_token + '\"' if auth_token else \"\"}\n",
" if use_vae:\n",
" vae_filename = vae_filename or vae_url.split(\"/\")[-1]\n",
" if not os.path.exists(f\"{models_path}/VAE/{vae_filename}\"):\n",
" ! curl -LJ {vae_url} -o {models_path}/VAE/{vae_filename} {'-H \"Authorization: Bearer ' + auth_token + '\"' if auth_token else \"\"}"
]
},
{
Expand All @@ -204,13 +222,12 @@
},
"outputs": [],
"source": [
"# @markdown &nbsp;\n",
"# @markdown ## Command line arguments\n",
"\n",
"import os\n",
"\n",
"no_half = False # @param {type:\"boolean\"}\n",
"no_half_vae = False # @param {type:\"boolean\"}\n",
"no_half_vae = True # @param {type:\"boolean\"}\n",
"allow_code = False # @param {type:\"boolean\"}\n",
"no_progressbar_hiding = False # @param {type:\"boolean\"}\n",
"medvram = False # @param {type:\"boolean\"}\n",
Expand All @@ -219,7 +236,7 @@
"xformers = True # @param {type:\"boolean\"}\n",
"disable_opt_split_attention = False # @param {type:\"boolean\"}\n",
"\n",
"# @markdown &nbsp;\n",
"# @markdown <br >\n",
"custom_arguments = \"\" # @param {type:\"string\"}\n",
"\n",
"run_string_with_variables = {\n",
Expand All @@ -232,22 +249,27 @@
" \"--deepdanbooru\": f\"{deepdanbooru}\",\n",
" \"--xformers\": f\"{xformers}\",\n",
" \"--disable-opt-split-attention\": f\"{disable_opt_split_attention}\",\n",
" \"--vae-path /content/stable-diffusion-webui/models/VAE/\" + vae_filename: f\"{use_vae}\",\n",
"}\n",
"\n",
"advanced_options = {k for (k, v) in run_string_with_variables.items() if v == \"True\"}\n",
"\n",
"# @markdown &nbsp;\n",
"# @markdown <br>\n",
"\n",
"# @markdown ## Enable password authentication (Prevent other users from using the WebUI)\n",
"\n",
"# @markdown &nbsp;\n",
"# @markdown <br >\n",
"\n",
"use_gradio_auth = False # @param {type:\"boolean\"}\n",
"gradio_auth_username = \"username\" # @param {type:\"string\"}\n",
"gradio_auth_password = \"password\" # @param {type:\"string\"}\n",
"\n",
"# @markdown &nbsp;\n",
"# @markdown <br>\n",
"\n",
"# @markdown # Advanced | Network preferences\n",
"\n",
"# @markdown &nbsp;\n",
"# @markdown <br>\n",
"\n",
"# @markdown ## Optional | Ngrok Tunnel\n",
"# @markdown Get token from [here](https://dashboard.ngrok.com/get-started/your-authtoken)\n",
"use_ngrok = False # @param {type: \"boolean\"}\n",
Expand All @@ -269,19 +291,22 @@
"if not ngrok_region or ngrok_region == \"default\":\n",
" ngrok_region = \"us\"\n",
"\n",
"# @markdown &nbsp;\n",
"# @markdown <br>\n",
"\n",
"# @markdown ## Optional | Tailscale\n",
"# @markdown Get auth key from [here](https://login.tailscale.com/admin/settings/keys)\n",
"use_tailscale = False # @param {type: \"boolean\"}\n",
"tailscale_auth_key = \"\" # @param {type: \"string\"}\n",
"\n",
"# @markdown &nbsp;\n",
"# @markdown <br>\n",
"\n",
"# @markdown ## Extensions\n",
"load_extensions_from_gdrive = True # @param {type:\"boolean\"}\n",
"extensions = \"https://github.com/yfszzx/stable-diffusion-webui-images-browser, https://github.com/DominikDoom/a1111-sd-webui-tagcomplete\" # @param {type:\"string\"}\n",
"extensions = list(map(str.strip, extensions.split(\",\")))\n",
"\n",
"# @markdown &nbsp;\n",
"# @markdown <br>\n",
"\n",
"# @markdown ## Save extensions to Google Drive\n",
"# @markdown **Deprecated** (Unexpected errors may occur)\n",
"save_extensions_to_gdrive = False # @param {type:\"boolean\"}\n",
Expand All @@ -305,6 +330,8 @@
" vars += \" --share\"\n",
"elif ngrok_auth_token and ngrok_region:\n",
" vars += f\" --ngrok {ngrok_auth_token} --ngrok-region {ngrok_region}\"\n",
"elif not ngrok_auth_token or not ngrok_region:\n",
" vars += \"--share\"\n",
"\n",
"if use_gradio_auth:\n",
" vars += f\" --gradio-auth {gradio_auth_username}:{gradio_auth_password}\"\n",
Expand Down Expand Up @@ -394,7 +421,7 @@
"python -c 'from launch import prepare_environment; prepare_environment()'\"\"\"\n",
"! {install_script}\n",
"\n",
"# @markdown &nbsp;\n",
"# @markdown \n",
"# @markdown ## Optional | Apply low RAM patch\n",
"apply_lowram_patch = True # @param {type: \"boolean\"}\n",
"\n",
Expand Down Expand Up @@ -441,7 +468,7 @@
"# @markdown keep in mind that this script is set to run for ever.\n",
"# @markdown > ※注意 このスクリプトは永久に実行されます。\n",
"\n",
"# @markdown &nbsp;\n",
"# @markdown \n",
"\n",
"# @markdown ### Important - click the public URL to launch WebUI in another tab\n",
"# @markdown > ### 重要 - 公開URLをクリックしてWebUIを起動してください\n",
Expand Down Expand Up @@ -475,8 +502,8 @@
"metadata": {
"accelerator": "GPU",
"colab": {
"provenance": [],
"include_colab_link": true
"include_colab_link": true,
"provenance": []
},
"gpuClass": "standard",
"kernelspec": {
Expand All @@ -486,7 +513,7 @@
},
"language_info": {
"name": "python",
"version": "3.10.8 | packaged by conda-forge | (main, Nov 24 2022, 14:07:00) [MSC v.1916 64 bit (AMD64)]"
"version": "3.10.0 | packaged by conda-forge | (default, Nov 10 2021, 13:20:59) [MSC v.1916 64 bit (AMD64)]"
},
"vscode": {
"interpreter": {
Expand All @@ -496,4 +523,4 @@
},
"nbformat": 4,
"nbformat_minor": 0
}
}

0 comments on commit fe5812f

Please sign in to comment.