jalr/tabbyAPI-ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Start: Migrate options from cu121/118 to cu12

Browse source 
This encapsulates more cuda versions and makes install easier for
new users.

Signed-off-by: kingbri <8082010+kingbri1@users.noreply.github.com>
This commit is contained in:
kingbri 2025-08-19 22:25:30 -04:00
parent 1344726936
commit 30a3cd75cf
6 changed files with 35 additions and 41 deletions
Download patch file Download diff file Expand all files Collapse all files

2
pyproject.toml
View file

@ -56,7 +56,7 @@ extras = [
dev = [
"ruff == 0.11.10"
]
cu121 = [
cu12 = [
# Torch (Extra index URLs not support in pyproject.toml)
"torch @ https://download.pytorch.org/whl/cu128/torch-2.8.0%2Bcu128-cp313-cp313-win_amd64.whl ; platform_system == 'Windows' and python_version == '3.13'",
"torch @ https://download.pytorch.org/whl/cu128/torch-2.8.0%2Bcu128-cp312-cp312-win_amd64.whl ; platform_system == 'Windows' and python_version == '3.12'",