soiz1 commited on
Commit
35d4e8a
·
verified ·
1 Parent(s): 7b75213

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -3,7 +3,8 @@ torchvision
3
  numpy
4
  rembg
5
  opencv-contrib-python
6
- flax
 
7
  onnxruntime
8
  diffusers==0.20.2
9
  transformers==4.29.2
 
3
  numpy
4
  rembg
5
  opencv-contrib-python
6
+ flax==0.7.2
7
+ jax>=0.4.1
8
  onnxruntime
9
  diffusers==0.20.2
10
  transformers==4.29.2