Spaces:
Build error
Build error
Khushi Dahiya
commited on
Commit
·
f628bd4
1
Parent(s):
8b07581
updating jax and flax versio
Browse files- requirements.txt +2 -2
requirements.txt
CHANGED
|
@@ -4,8 +4,8 @@ python-multipart==0.0.6
|
|
| 4 |
numpy==1.24.3
|
| 5 |
pydantic==2.5.0
|
| 6 |
python-jose[cryptography]==3.3.0
|
| 7 |
-
jax[cuda12_pip]
|
| 8 |
-
flax
|
| 9 |
tf-nightly
|
| 10 |
tensorflow-text-nightly
|
| 11 |
tf2jax==0.3.8
|
|
|
|
| 4 |
numpy==1.24.3
|
| 5 |
pydantic==2.5.0
|
| 6 |
python-jose[cryptography]==3.3.0
|
| 7 |
+
jax[cuda12_pip]>=0.7.1
|
| 8 |
+
flax>=0.7.5
|
| 9 |
tf-nightly
|
| 10 |
tensorflow-text-nightly
|
| 11 |
tf2jax==0.3.8
|