download standard model via torrent option

This commit is contained in:
Engineer-of-Stuff 2022-09-08 19:55:21 -06:00 committed by GitHub
parent 611e6cbe0c
commit 452235a9ae
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 66 additions and 27 deletions

View File

@ -14,6 +14,7 @@
{
"cell_type": "markdown",
"metadata": {
"jp-MarkdownHeadingCollapsed": true,
"tags": []
},
"source": [
@ -46,7 +47,8 @@
"cell_type": "code",
"execution_count": null,
"metadata": {
"id": "sBbcB4vwj_jm"
"id": "sBbcB4vwj_jm",
"tags": []
},
"outputs": [],
"source": [
@ -120,6 +122,10 @@
"cell_type": "markdown",
"metadata": {},
"source": [
"I've provided a few different ways of aquiring the models. Try the torrent option first.\n",
"\n",
"I recommend downloading both the standard and waifu diffusion. You can choose which one you want later.\n",
"\n",
"You don't need to repeat this step if you've already downloaded the models."
]
},
@ -137,6 +143,28 @@
"cell_type": "markdown",
"metadata": {},
"source": [
"**Torrent**"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"!apt update\n",
"!apt install -y aria2\n",
"%cd /notebooks/models\n",
"!aria2c --seed-time=0 --max-overall-upload-limit=1K \"magnet:?xt=urn:btih:3A4A612D75ED088EA542ACAC52F9F45987488D1C&tr=udp://tracker.opentrackr.org:1337\"\n",
"!rm /notebooks/models/sd-v1-4.ckpt.aria2"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"**Web Download**\n",
"\n",
"Voldy provided an alternative download if you don't want to use HuggingFace.\n",
"\n",
"[https://drive.google.com/file/d/1wHFgl0ivCmIZv88hVZXkb8oy9qCuaBGA/view](https://drive.google.com/file/d/1wHFgl0ivCmIZv88hVZXkb8oy9qCuaBGA/view)\n",
@ -154,7 +182,7 @@
},
"outputs": [],
"source": [
"user_token = \"<enter user_token here>\"\n",
"user_token = \"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\"\n",
"\n",
"# ===============================================================================================\n",
"\n",
@ -172,6 +200,26 @@
"#### Waifu Diffusion"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"**Torrent**"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"!apt update\n",
"!apt install -y aria2\n",
"%cd /notebooks/models\n",
"!aria2c --seed-time=0 --max-overall-upload-limit=1K \"magnet:?xt=urn:btih:F45CECF4E9DE86DA83A78DD2CCCD7F27D5557A52&tr=udp://nyquist.localghost.org:6969\"\n",
"!rm /notebooks/models/wd-v1-2-full-ema.ckpt.aria2"
]
},
{
"cell_type": "markdown",
"metadata": {
@ -192,28 +240,6 @@
"!wget https://thisanimedoesnotexist.ai/downloads/wd-v1-2-full-ema.ckpt -O /notebooks/models/wd-v1-2-full-ema.ckpt"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"**Torrent**\n",
"\n",
"Very fast."
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"!apt update\n",
"!apt install -y aria2\n",
"%cd /notebooks/models\n",
"!aria2c --seed-time=0 --max-overall-upload-limit=1K \"magnet:?xt=urn:btih:F45CECF4E9DE86DA83A78DD2CCCD7F27D5557A52&tr=udp://nyquist.localghost.org:6969\"\n",
"!rm /notebooks/models/wd-v1-2-full-ema.ckpt.aria2"
]
},
{
"cell_type": "markdown",
"metadata": {
@ -264,7 +290,9 @@
"source": [
"First, enter the name of the model you want to use into `model_name` in the cell below.\n",
"\n",
"Then run the model. You will get a link to nnn.gradio.app, that's your WebUI. Follow it."
"Then run the model. You will get a link to nnn.gradio.app, that's your WebUI. Follow it.\n",
"\n",
"If you have any issues, restart the kernel."
]
},
{
@ -278,8 +306,8 @@
"outputs": [],
"source": [
"# Possible models:\n",
"# wd-v1-2-full-ema.ckpt (waifu-diffusion v1.2)\n",
"# sd-v1-4.ckpt (standard diffusion v1.4)\n",
"# wd-v1-2-full-ema.ckpt (waifu-diffusion v1.2)\n",
"\n",
"model_name = ''\n",
"\n",
@ -293,9 +321,20 @@
"if not os.path.exists(model_file):\n",
" print('Cannot find model file:', model_file)\n",
" sys.exit(1)\n",
" \n",
"\n",
"if model_name == '':\n",
" print('You forgot to fill out the variable model_file above.')\n",
" sys.exit(1)\n",
"\n",
"!python /notebooks/stable-diffusion-webui/webui.py --share --ckpt $model_file"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": []
}
],
"metadata": {