From 5cc7d0038fe7de839c71cb1db2bfe5e3194dfe30 Mon Sep 17 00:00:00 2001
From: Brett Kuprel <brkuprel@gmail.com>
Date: Fri, 1 Jul 2022 11:32:36 -0400
Subject: [PATCH] fixed setup_flax.sh

---
 setup_flax.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup_flax.sh b/setup_flax.sh
index ee465bd..0376eed 100644
--- 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