Commit
·
3c8ec9c
1
Parent(s):
fe53c41
Update requirements.txt
Browse files- requirements.txt +5 -1
requirements.txt
CHANGED
|
@@ -1,3 +1,7 @@
|
|
| 1 |
transformers==4.29.2
|
| 2 |
tensorflow==2.12.0
|
| 3 |
-
torch
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
transformers==4.29.2
|
| 2 |
tensorflow==2.12.0
|
| 3 |
+
torch
|
| 4 |
+
jax>=0.2.8
|
| 5 |
+
jaxlib>=0.1.59
|
| 6 |
+
flax>=0.3.5
|
| 7 |
+
optax>=0.0.9
|