fixed setup_flax.sh

2022-07-01 11:32:36 -04:00
parent 2a4e3db15b
commit 5cc7d0038f
1 changed files with 1 additions and 1 deletions
--- a/setup_flax.sh
+++ b/setup_flax.sh
@@ -2,7 +2,7 @@

 set -e

-pip install -r requirements.txt
+pip install torch flax wandb

 # download vqgan
 mkdir -p pretrained/vqgan