mirror of
https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git
synced 2025-08-11 22:18:57 +08:00
Added onnxruntime-gpu (#4631)
### What problem does this PR solve? Added onnxruntime-gpu ### Type of change - [x] Refactoring
This commit is contained in:
parent
898ae7fa80
commit
8b49734241
@ -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",
|
||||
@ -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…
x
Reference in New Issue
Block a user