Update requirements.txt
Browse files- requirements.txt +1 -1
requirements.txt
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
uvicorn==0.17.1
|
| 2 |
orbax==0.1.5
|
| 3 |
pydantic==1.*
|
| 4 |
-
jax[cpu]
|
| 5 |
flax==0.6.6
|
| 6 |
diffusers>=0.13.1
|
| 7 |
accelerate==0.16.0
|
|
|
|
| 1 |
uvicorn==0.17.1
|
| 2 |
orbax==0.1.5
|
| 3 |
pydantic==1.*
|
| 4 |
+
jax[cpu]>=0.4.4
|
| 5 |
flax==0.6.6
|
| 6 |
diffusers>=0.13.1
|
| 7 |
accelerate==0.16.0
|