zhichyu commited on
Commit
c5006c2
·
1 Parent(s): d453e49

Removed onnxruntime (#4632)

Browse files

### What problem does this PR solve?

Removed onnxruntime. It conflicts with the onnxruntime-gpu.

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)

Files changed (2) hide show
  1. pyproject.toml +6 -5
  2. uv.lock +0 -0
pyproject.toml CHANGED
@@ -59,8 +59,8 @@ dependencies = [
59
  "nltk==3.9.1",
60
  "numpy>=1.26.0,<2.0.0",
61
  "ollama==0.2.1",
62
- "onnxruntime==1.19.2",
63
- "onnxruntime-gpu==1.19.2",
64
  "openai==1.45.0",
65
  "opencv-python==4.10.0.84",
66
  "opencv-python-headless==4.10.0.84",
@@ -127,9 +127,10 @@ dependencies = [
127
 
128
  [project.optional-dependencies]
129
  full = [
130
- "bcembedding>=0.1.5",
131
- "fastembed>=0.3.6,<0.4.0",
132
- "flagembedding>=1.2.10,<2.0.0",
 
133
  "torch>=2.5.0,<3.0.0",
134
  "transformers>=4.35.0,<5.0.0"
135
  ]
 
59
  "nltk==3.9.1",
60
  "numpy>=1.26.0,<2.0.0",
61
  "ollama==0.2.1",
62
+ "onnxruntime==1.19.2; sys_platform == 'darwin' or platform_machine == 'arm64'",
63
+ "onnxruntime-gpu==1.19.2; platform_machine == 'x86_64'",
64
  "openai==1.45.0",
65
  "opencv-python==4.10.0.84",
66
  "opencv-python-headless==4.10.0.84",
 
127
 
128
  [project.optional-dependencies]
129
  full = [
130
+ "bcembedding==0.1.5",
131
+ "fastembed>=0.3.6,<0.4.0; sys_platform == 'darwin' or platform_machine == 'arm64'",
132
+ "fastembed-gpu>=0.3.6,<0.4.0; platform_machine == 'x86_64'",
133
+ "flagembedding==1.2.10",
134
  "torch>=2.5.0,<3.0.0",
135
  "transformers>=4.35.0,<5.0.0"
136
  ]
uv.lock CHANGED
The diff for this file is too large to render. See raw diff