LittleFrog commited on
Commit
9d45664
·
verified ·
1 Parent(s): dbe1190

Use rembg CPU mode. Load with Onnx-gpu will cause ZeroGpu to fail.

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -28,7 +28,7 @@ lovely-tensors>=0.1.14
28
  plotly==5.13.1
29
  tyro==0.3.38
30
  ray
31
- rembg==2.0.67
32
  onnxruntime-gpu==1.22.0
33
  pydantic==2.10.6
34
  spaces
 
28
  plotly==5.13.1
29
  tyro==0.3.38
30
  ray
31
+ rembg[cpu]==2.0.67
32
  onnxruntime-gpu==1.22.0
33
  pydantic==2.10.6
34
  spaces