From 41ec1ab161fa07af3852ecd1d4a7b387dafcf09a Mon Sep 17 00:00:00 2001
From: Victor Hall <victor.charles.hall@gmail.com>
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. "
    ]
   },
   {