From 6ec70b69e1ac77473fd73c88fcde19e7b77cf0ac Mon Sep 17 00:00:00 2001
From: Victor Hall <victor.charles.hall@gmail.com>
Date: Wed, 23 Nov 2022 21:46:47 -0500
Subject: [PATCH] update notebook

---
 Train-Runpod.ipynb | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/Train-Runpod.ipynb b/Train-Runpod.ipynb
index 85d3034..ccdb83a 100644
--- a/Train-Runpod.ipynb
+++ b/Train-Runpod.ipynb
@@ -143,6 +143,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "# Default is 1.5 with the new VAE, you can change this to another file on huggingface if you want:\n",
     "from huggingface_hub import hf_hub_download\n",
     "downloaded_model_path = hf_hub_download(\n",
     " repo_id=\"panopstor/EveryDream\",\n",
@@ -234,7 +235,7 @@
    "outputs": [],
    "source": [
     "# run the trainer, wait until it finishes then SCROLL DOWN to the next cell\n",
-    "!python main.py --base {my_yaml} -t  --actual_resume {downloaded_model_path} -n test --data_root input\n",
+    "!python main.py --base {my_yaml} -t  --actual_resume whatever.ckpt -n test --data_root input\n",
     "print(finish_msg)"
    ]
   },