Added onnxruntime-gpu (#4631)

### What problem does this PR solve?

Added onnxruntime-gpu

### Type of change

- [x] Refactoring
This commit is contained in:
Zhichang Yu 2025-01-24 14:33:21 +08:00 committed by GitHub
parent 898ae7fa80
commit 8b49734241
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3064 additions and 3071 deletions

View File

@ -60,6 +60,7 @@ dependencies = [
"numpy>=1.26.0,<2.0.0", "numpy>=1.26.0,<2.0.0",
"ollama==0.2.1", "ollama==0.2.1",
"onnxruntime==1.19.2", "onnxruntime==1.19.2",
"onnxruntime-gpu==1.19.2",
"openai==1.45.0", "openai==1.45.0",
"opencv-python==4.10.0.84", "opencv-python==4.10.0.84",
"opencv-python-headless==4.10.0.84", "opencv-python-headless==4.10.0.84",
@ -126,9 +127,9 @@ dependencies = [
[project.optional-dependencies] [project.optional-dependencies]
full = [ full = [
"bcembedding==0.1.3", "bcembedding>=0.1.5",
"fastembed>=0.3.6,<0.4.0", "fastembed>=0.3.6,<0.4.0",
"flagembedding==1.2.10", "flagembedding>=1.2.10,<2.0.0",
"torch==2.3.0", "torch>=2.5.0,<3.0.0",
"transformers==4.38.1" "transformers>=4.35.0,<5.0.0"
] ]

6126
uv.lock generated

File diff suppressed because it is too large Load Diff