mirror of
https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git
synced 2025-08-12 02:19:05 +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",
|
"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"
|
||||||
]
|
]
|
Loading…
x
Reference in New Issue
Block a user