Mirrors
/
stable-diffusion-paperspace
mirror of https://github.com/Engineer-of-Stuff/stable-diffusion-paperspace.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add files via upload

This commit is contained in:
Cyberes 2022-09-18 00:25:08 -06:00 committed by GitHub
parent ba6e9350e6
commit 4ad2c1541a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 46 additions and 92 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

138
StableDiffusionUI_Voldemort_paperspace.ipynb
View File

@ -43,7 +43,7 @@
"\n", "\n",
"`/notebooks/` is storage for this notebook only.\n", "`/notebooks/` is storage for this notebook only.\n",
"\n", "\n",
"We're going to store models in `/storage/models`.\n", "We're going to store models in `/storage/models` and create a symlink.\n",
"\n", "\n",
"_You must run the block below or else the variable won't be accessable in the notebook._" "_You must run the block below or else the variable won't be accessable in the notebook._"
] ]
@ -59,8 +59,7 @@
"# ===============================================================\n", "# ===============================================================\n",
"# Save the variable to Jupiter's temp storage to access it when the kernel restarts.\n", "# Save the variable to Jupiter's temp storage to access it when the kernel restarts.\n",
"# To reset your storage directory, rerun this cell.\n", "# To reset your storage directory, rerun this cell.\n",
"%store model_storage_dir\n", "%store model_storage_dir"
"!mkdir -p $model_storage_dir"
] ]
}, },
{ {
@ -254,7 +253,6 @@
{ {
"cell_type": "markdown", "cell_type": "markdown",
"metadata": { "metadata": {
"jp-MarkdownHeadingCollapsed": true,
"tags": [] "tags": []
}, },
"source": [ "source": [
@ -270,23 +268,6 @@
"The 60000 steps version is the original, the 115000 steps is the 60000 with additional training. Use the 60000 step version if the style nudging is too much." "The 60000 steps version is the original, the 115000 steps is the 60000 with additional training. Use the 60000 step version if the style nudging is too much."
] ]
}, },
{
"cell_type": "markdown",
"metadata": {},
"source": [
"**115000**"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"%store -r model_storage_dir\n",
"!wget https://huggingface.co/naclbit/trinart_stable_diffusion_v2/resolve/main/trinart2_step115000.ckpt -O $model_storage_dir/trinart2_step115000.ckpt"
]
},
{ {
"cell_type": "markdown", "cell_type": "markdown",
"metadata": {}, "metadata": {},
@ -304,23 +285,11 @@
"!wget https://huggingface.co/naclbit/trinart_stable_diffusion_v2/resolve/main/trinart2_step60000.ckpt -O $model_storage_dir/trinart2_step60000.ckpt" "!wget https://huggingface.co/naclbit/trinart_stable_diffusion_v2/resolve/main/trinart2_step60000.ckpt -O $model_storage_dir/trinart2_step60000.ckpt"
] ]
}, },
{
"cell_type": "markdown",
"metadata": {
"jp-MarkdownHeadingCollapsed": true,
"tags": []
},
"source": [
"#### Gradient Dataset Import"
]
},
{ {
"cell_type": "markdown", "cell_type": "markdown",
"metadata": {}, "metadata": {},
"source": [ "source": [
"Paperspace provides a way to mount external storage to notebooks, called *Datasets*. This is an advanced method but it will save you many GB since you can reuse the same model file across different notebooks.\n", "**95000**"
"\n",
"https://docs.paperspace.com/gradient/cli/datasets"
] ]
}, },
{ {
@ -329,15 +298,25 @@
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
"source": [ "source": [
"mounted_dataset_path = '/datasets/<name>'\n",
"\n",
"# ============================================================\n",
"%store -r model_storage_dir\n", "%store -r model_storage_dir\n",
"import os\n", "!wget https://huggingface.co/naclbit/trinart_stable_diffusion_v2/resolve/main/trinart2_step95000.ckpt -O $model_storage_dir/trinart2_step95000.ckpt"
"for file in os.listdir(mounted_dataset_path):\n", ]
" if file.endswith(\"ckpt\"):\n", },
" !ln -s $mounted_dataset_path/$file $model_storage_dir/$file\n", {
" !ls -la --block-size=GB $model_storage_dir/$file" "cell_type": "markdown",
"metadata": {},
"source": [
"**115000**"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"%store -r model_storage_dir\n",
"!wget https://huggingface.co/naclbit/trinart_stable_diffusion_v2/resolve/main/trinart2_step115000.ckpt -O $model_storage_dir/trinart2_step115000.ckpt"
] ]
}, },
{ {
@ -368,6 +347,29 @@
"os.kill(os.getpid(), 9)" "os.kill(os.getpid(), 9)"
] ]
}, },
{
"cell_type": "markdown",
"metadata": {},
"source": [
"### Link the models directory\n",
"\n",
"If it says `failed to create symbolic link: File exists` that's fine. It means you've already ran this block."
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"%store -r model_storage_dir\n",
"import os\n",
"for file in os.listdir(model_storage_dir):\n",
" if file.endswith(\"ckpt\"):\n",
" !ln -s $model_storage_dir/$file /notebooks/stable-diffusion-webui/models/$file\n",
" !ls -la --block-size=GB /notebooks/stable-diffusion-webui/models/$file"
]
},
{ {
"cell_type": "markdown", "cell_type": "markdown",
"metadata": { "metadata": {
@ -381,39 +383,11 @@
"cell_type": "markdown", "cell_type": "markdown",
"metadata": {}, "metadata": {},
"source": [ "source": [
"First, enter the name of the model you want to use into `model_name` in the cell below.\n", "Run this block to launch the WebUI. You will get a link to nnn.gradio.app, that's your WebUI. Follow it.\n",
"\n",
"Then run the model. You will get a link to nnn.gradio.app, that's your WebUI. Follow it.\n",
"\n", "\n",
"If you have any issues, restart the kernel." "If you have any issues, restart the kernel."
] ]
}, },
{
"cell_type": "markdown",
"metadata": {},
"source": [
"**Choose your Model**\n",
"\n",
"|Name||\n",
"|-|-|\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",
"|`trinart2_step115000.ckpt`|trinart2 v2 11500 steps|\n",
"|`trinart2_step60000.ckpt`|trinart2 v2 60000 steps|\n",
"\n",
"Enter the name of the model you want to use in `model_name` below.\n",
"\n",
"To change models you must restart the kernel for it to take effect."
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"**Start the WebUI**"
]
},
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": null, "execution_count": null,
@ -423,28 +397,8 @@
}, },
"outputs": [], "outputs": [],
"source": [ "source": [
"model_name = ''\n",
"\n",
"# ===============================================================================================================\n",
"# Don't change anything in here\n",
"%store -r model_storage_dir\n", "%store -r model_storage_dir\n",
"model_file = f'{model_storage_dir}/{model_name}'\n", "!python /notebooks/stable-diffusion-webui/webui.py --share # --opt-split-attention"
"\n",
"import sys\n",
"import os\n",
"\n",
"if not os.path.exists(model_file):\n",
" print('Cannot find model file:', model_file)\n",
" sys.exit(1)\n",
"\n",
"if model_name == '':\n",
" print('You forgot to fill out the variable model_name above.')\n",
" sys.exit(1)\n",
"\n",
"# ===============================================================================================================\n",
" \n",
"# You can modify the args if nessesary\n",
"!python /notebooks/stable-diffusion-webui/webui.py --share --ckpt $model_file # --opt-split-attention"
] ]
}, },
{ {