Skip to content

Commit

Permalink
Added onnxruntime-gpu (#4631)
Browse files Browse the repository at this point in the history
### What problem does this PR solve?

Added onnxruntime-gpu

### Type of change

- [x] Refactoring
  • Loading branch information
yuzhichang authored Jan 24, 2025
1 parent 898ae7f commit 8b49734
Show file tree
Hide file tree
Showing 2 changed files with 3,121 additions and 3,128 deletions.
9 changes: 5 additions & 4 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -60,6 +60,7 @@ dependencies = [
"numpy>=1.26.0,<2.0.0",
"ollama==0.2.1",
"onnxruntime==1.19.2",
"onnxruntime-gpu==1.19.2",
"openai==1.45.0",
"opencv-python==4.10.0.84",
"opencv-python-headless==4.10.0.84",
Expand Down Expand Up @@ -126,9 +127,9 @@ dependencies = [

[project.optional-dependencies]
full = [
"bcembedding==0.1.3",
"bcembedding>=0.1.5",
"fastembed>=0.3.6,<0.4.0",
"flagembedding==1.2.10",
"torch==2.3.0",
"transformers==4.38.1"
"flagembedding>=1.2.10,<2.0.0",
"torch>=2.5.0,<3.0.0",
"transformers>=4.35.0,<5.0.0"
]
Loading

0 comments on commit 8b49734

Please sign in to comment.