From 41ec1ab161fa07af3852ecd1d4a7b387dafcf09a Mon Sep 17 00:00:00 2001 From: Victor Hall Date: Wed, 23 Nov 2022 16:53:45 -0500 Subject: [PATCH] minor update to runpod --- Train-Runpod.ipynb | 53 ++++++++++++++++++++++++++++++++++++++++++---- 1 file changed, 49 insertions(+), 4 deletions(-) diff --git a/Train-Runpod.ipynb b/Train-Runpod.ipynb index 9e979f5..85d3034 100644 --- a/Train-Runpod.ipynb +++ b/Train-Runpod.ipynb @@ -145,8 +145,8 @@ "source": [ "from huggingface_hub import hf_hub_download\n", "downloaded_model_path = hf_hub_download(\n", - " repo_id=\"runwayml/stable-diffusion-v1-5\",\n", - " filename=\"v1-5-pruned-emaonly.ckpt\",\n", + " repo_id=\"panopstor/EveryDream\",\n", + " filename=\"sd_v1-5_vae.ckpt\",\n", " use_auth_token=True\n", ")\n", "print(downloaded_model_path) # cache location\n", @@ -244,7 +244,9 @@ "metadata": {}, "source": [ "## Prune your checkpoints\n", - "This will create 2GB pruned files for all your checkpoints and delete the 11GB files." + "This will create 2GB pruned files for all your checkpoints and delete the 11GB files.\n", + "\n", + "If you wish to resume, you may want to remove \"--delete\" command below and manually delete the 11GB files you dont want to keep. Typically you only need the last 1 checkpoint in 11gb for resuming." ] }, { @@ -258,6 +260,47 @@ "!python scripts/autoprune_all.py --delete" ] }, + { + "cell_type": "markdown", + "id": "065dfd91", + "metadata": {}, + "source": [ + "## Test\n", + "Look in the file drawer on the left for your epoch ckpt names and try them out in the cell below one at a time. You can save time just downloading the one pruned file that looks the best. Try each out.\n", + "\n", + "Change the prompt and the ckpt_path below to appropriate values." + ] + }, + { + "cell_type": "code", + "execution_count": null, + "id": "954dfc20", + "metadata": {}, + "outputs": [], + "source": [ + "!python scripts/txt2img.py --ckpt_path \"epoch=02-step=01000.ckpt\" \\\n", + "--n_samples 2 \\\n", + "--n_iter 4 \\\n", + "--prompt \"a boy and his dog talking a walk down the sidewalk\" \\\n", + "--scale 6.0 \\\n", + "--outdir outputs \\" + ] + }, + { + "cell_type": "code", + "execution_count": null, + "id": "e13533d6", + "metadata": {}, + "outputs": [], + "source": [ + "# run to show the images here, or use the file drawer on the left to look at them\n", + "import glob\n", + "from IPython.display import Image, display\n", + "for imageName in glob.glob('outputs/*.png'): #assuming JPG\n", + " display(Image(filename=imageName))\n", + " print(imageName)" + ] + }, { "cell_type": "markdown", "id": "51456afe", @@ -267,6 +310,8 @@ "\n", "Use the cell below to generate links, right click and save as to download. If you use Colab, you can skip this and use the Gdrive connect instead.\n", "\n", + "If the links don't work, you can double left click the ckpt file in the file drawer on the left, then go to \"File\" menu then \"Download\".\n", + "\n", "[EveryDream Discord](https://discord.gg/uheqxU6sXN)" ] }, @@ -291,7 +336,7 @@ "source": [ "# Gdrive connect\n", "\n", - "For colab only, copies your ckpts to your gdrive. If the EveryDreamCkpts folder already exists in your gdrive there will be an error, but it should still copy your files. " + "For Colab only, copies your ckpts to your gdrive. If the EveryDreamCkpts folder already exists in your gdrive there will be an error, but it should still copy your files. " ] }, {