add back requirements.txt

README.md

@@ -9,7 +9,7 @@ It currently takes **7.4 seconds** to generate an image with DALL·E Mega with P
 
 ### Setup
 
-Run either `sh setup_torch.sh` or `sh setup_flax.sh` to install dependencies and download pretrained models.  The torch models can be manually downloaded [here](https://huggingface.co/kuprel/min-dalle/tree/main).
+Run either `sh setup.sh` to install dependencies and download pretrained models.  The torch models can be manually downloaded [here](https://huggingface.co/kuprel/min-dalle/tree/main).
 The flax models can be manually downloaded here: 
 [VQGan](https://huggingface.co/dalle-mini/vqgan_imagenet_f16_16384), 
 [DALL·E Mini](https://wandb.ai/dalle-mini/dalle-mini/artifacts/DalleBart_model/mini-1/v0/files), 

requirements.txt

@@ -0,0 +1 @@
+torch

requirements_flax.txt

@@ -0,0 +1,3 @@
+flax
+torch
+wandb

setup.sh

@@ -2,7 +2,7 @@
 
 set -e
 
-pip3 install torch
+pip3 install -r requirements.txt
 
 mkdir -p ./pretrained/dalle_bart_mega/
 curl https://huggingface.co/kuprel/min-dalle/resolve/main/vocab.json -L --output ./pretrained/dalle_bart_mega/vocab.json

setup_flax.sh

@@ -2,7 +2,7 @@
 
 set -e
 
-pip install torch flax wandb
+pip3 install -r requirements_flax.txt
 
 # download vqgan
 mkdir -p pretrained/vqgan