cleanup

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
 

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"
       ]
     },
     {