mega works with latest flax version 0.5.2 now, removing 0.4.2 pin

This commit is contained in:

Brett Kuprel

2022-07-01 02:58:43 -04:00

parent eaee59a1ef

commit b40fd83a0d

5 changed files with 25 additions and 27 deletions

requirements.txt

Vendored

+1 -1

View File

@@ -1,3 +1,3 @@
 torch
 flax==0.4.2
 flax
 wandb