works with latest flax version 0.5.2, updated requirements.txt

2022-06-29 11:46:19 -04:00
parent c4f613c89f
commit 764b5bbc0e
3 changed files with 5 additions and 5 deletions
@@ -1,3 +1,3 @@
 torch
-flax==0.4.2
+flax==0.5.2
 wandb