Added onnxruntime-gpu (#4631)
Browse files### What problem does this PR solve?
Added onnxruntime-gpu
### Type of change
- [x] Refactoring
- pyproject.toml +5 -4
- uv.lock +0 -0
pyproject.toml
CHANGED
|
@@ -60,6 +60,7 @@ dependencies = [
|
|
| 60 |
"numpy>=1.26.0,<2.0.0",
|
| 61 |
"ollama==0.2.1",
|
| 62 |
"onnxruntime==1.19.2",
|
|
|
|
| 63 |
"openai==1.45.0",
|
| 64 |
"opencv-python==4.10.0.84",
|
| 65 |
"opencv-python-headless==4.10.0.84",
|
|
@@ -126,9 +127,9 @@ dependencies = [
|
|
| 126 |
|
| 127 |
[project.optional-dependencies]
|
| 128 |
full = [
|
| 129 |
-
"bcembedding
|
| 130 |
"fastembed>=0.3.6,<0.4.0",
|
| 131 |
-
"flagembedding
|
| 132 |
-
"torch
|
| 133 |
-
"transformers
|
| 134 |
]
|
|
|
|
| 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 |
|
| 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 |
]
|
uv.lock
CHANGED
|
The diff for this file is too large to render.
See raw diff
|
|
|