浏览代码

Added onnxruntime-gpu (#4631)

### What problem does this PR solve?

Added onnxruntime-gpu

### Type of change

- [x] Refactoring
tags/v0.16.0
Zhichang Yu 9 个月前
父节点
当前提交
8b49734241
没有帐户链接到提交者的电子邮件
共有 2 个文件被更改,包括 3121 次插入3128 次删除
  1. 5
    4
      pyproject.toml
  2. 3116
    3124
      uv.lock

+ 5
- 4
pyproject.toml 查看文件

@@ -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"
]

+ 3116
- 3124
uv.lock
文件差异内容过多而无法显示
查看文件


正在加载...
取消
保存