half size wd prined

Former-commit-id: 2c33261590
This commit is contained in:
Engineer-of-Stuff 2022-09-09 12:40:33 -06:00 committed by GitHub
parent e8fceff665
commit bda7211175
1 changed files with 24 additions and 0 deletions

View File

@ -249,9 +249,32 @@
"!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": [
"**Half-size Model**\n",
"\n",
"If you're on free tier try this first."
]
},
{
"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:153590FD7E93EE11D8DB951451056C362E3A9150&dn=wd-v1-2-full-ema-pruned.ckpt&tr=udp://glotorrents.pw:6969/announce&tr=udp://tracker.opentrackr.org:1337/announce&tr=udp://torrent.gresille.org:80/announce&tr=udp://tracker.openbittorrent.com:80&tr=udp://tracker.coppersurfer.tk:6969&tr=udp://tracker.leechers-paradise.org:6969&tr=udp://p4p.arenabg.ch:1337&tr=udp://tracker.internetwarriors.net:1337&tr=wss://tracker.openwebtorrent.com\"\n",
"!rm /notebooks/models/wd-v1-2-full-ema.ckpt.aria2"
]
},
{
"cell_type": "markdown",
"metadata": {
"jp-MarkdownHeadingCollapsed": true,
"tags": []
},
"source": [
@ -333,6 +356,7 @@
"# Possible models:\n",
"# sd-v1-4.ckpt (standard diffusion v1.4)\n",
"# wd-v1-2-full-ema.ckpt (waifu-diffusion v1.2)\n",
"# wd-v1-2-full-ema-pruned.ckpt (waifu-diffusion v1.2 half-size)\n",
"\n",
"model_name = ''\n",
"\n",