fixed setup_flax.sh

2 years ago · 5cc7d0038f
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