add back requirements.txt

2022-07-01 12:53:55 -04:00
parent f71bd054fd
commit b61041f766
5 changed files with 7 additions and 3 deletions
@@ -2,7 +2,7 @@

 set -e

-pip install torch flax wandb
+pip3 install -r requirements_flax.txt

 # download vqgan
 mkdir -p pretrained/vqgan