From 2097c6ed5a2f5e84fa48e1df12ff58c0f0f28a40 Mon Sep 17 00:00:00 2001
From: Brett Kuprel <brkuprel@gmail.com>
Date: Fri, 1 Jul 2022 14:35:13 -0400
Subject: [PATCH] cleanup

---
 README.md       | 2 +-
 min_dalle.ipynb | 7 +++----
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/README.md b/README.md
index ebe08fd..dc8d834 100644
--- a/README.md
+++ b/README.md
@@ -7,7 +7,7 @@ This is a minimal implementation of Boris Dayma's [DALL·E Mini](https://github.
 
 It currently takes **7.4 seconds** to generate an image with DALL·E Mega with PyTorch on a standard GPU runtime in Colab
 
-The flax model, and the code for coverting it to torch, has been moved [here](https://github.com/kuprel/min-dalle-flax).
+The flax model, and the code for coverting it to torch, have been moved [here](https://github.com/kuprel/min-dalle-flax).
 
 ### Setup
 
diff --git a/min_dalle.ipynb b/min_dalle.ipynb
index daaf73d..c0bab7e 100644
--- a/min_dalle.ipynb
+++ b/min_dalle.ipynb
@@ -84,20 +84,19 @@
         "%%shell\n",
         "\n",
         "repo_path=\"https://huggingface.co/kuprel/min-dalle/resolve/main\"\n",
-        "\n",
         "mega_path=\"/content/min-dalle/pretrained/dalle_bart_mega\"\n",
         "vqgan_path=\"/content/min-dalle/pretrained/vqgan\"\n",
         "\n",
         "git clone https://github.com/kuprel/min-dalle\n",
         "\n",
-        "mkdir -p ${vqgan_path}\n",
         "mkdir -p ${mega_path}\n",
+        "mkdir -p ${vqgan_path}\n",
         "\n",
-        "curl ${repo_path}/detoker.pt -L --output ${vqgan_path}/detoker.pt\n",
         "curl ${repo_path}/vocab.json -L --output ${mega_path}/vocab.json\n",
         "curl ${repo_path}/merges.txt -L --output ${mega_path}/merges.txt\n",
         "curl ${repo_path}/encoder.pt -L --output ${mega_path}/encoder.pt\n",
-        "curl ${repo_path}/decoder.pt -L --output ${mega_path}/decoder.pt"
+        "curl ${repo_path}/decoder.pt -L --output ${mega_path}/decoder.pt\n",
+        "curl ${repo_path}/detoker.pt -L --output ${vqgan_path}/detoker.pt"
       ]
     },
     {