LittleFrog commited on
Commit
6109299
·
1 Parent(s): 3c28ba8

install onnx gpu

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -28,7 +28,8 @@ lovely-tensors>=0.1.14
28
  plotly==5.13.1
29
  tyro==0.3.38
30
  ray
31
- rembg
 
32
  spaces
33
  loguru
34
  -e git+https://github.com/CompVis/taming-transformers.git#egg=taming-transformers
 
28
  plotly==5.13.1
29
  tyro==0.3.38
30
  ray
31
+ rembg==2.0.67
32
+ onnxruntime-gpu==1.22.0
33
  spaces
34
  loguru
35
  -e git+https://github.com/CompVis/taming-transformers.git#egg=taming-transformers