Просмотр исходного кода

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 месяцев назад
Родитель
Сommit
8b49734241
Аккаунт пользователя с таким Email не найден
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
Разница между файлами не показана из-за своего большого размера
Просмотреть файл


Загрузка…
Отмена
Сохранить