Update requirements.txt
Browse files- requirements.txt +2 -1
requirements.txt
CHANGED
|
@@ -1,4 +1,5 @@
|
|
| 1 |
uvicorn==0.17.1
|
|
|
|
| 2 |
pydantic==1.*
|
| 3 |
jax[cpu]==0.4.4
|
| 4 |
flax==0.6.6
|
|
@@ -8,5 +9,5 @@ transformers==4.26.1
|
|
| 8 |
huggingface-hub>=0.15.0
|
| 9 |
torch==1.13.1
|
| 10 |
safetensors==0.2.8
|
| 11 |
-
|
| 12 |
|
|
|
|
| 1 |
uvicorn==0.17.1
|
| 2 |
+
orbax==0.1.5
|
| 3 |
pydantic==1.*
|
| 4 |
jax[cpu]==0.4.4
|
| 5 |
flax==0.6.6
|
|
|
|
| 9 |
huggingface-hub>=0.15.0
|
| 10 |
torch==1.13.1
|
| 11 |
safetensors==0.2.8
|
| 12 |
+
|
| 13 |
|