opensource/WeClone-new
3
0
Fork 0
mirror of https://github.com/xming521/WeClone.git synced 2026-08-29 02:10:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

更新pyproject.toml中的torch版本至2.7.0，并将pytorch源从cu124更新至cu126。

Browse Source
This commit is contained in:
xming521
2025-06-14 22:24:19 +08:00
parent f3c2fa6ef7
commit 1f43e6325f
1 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pyproject.toml
+9 -9
View File
@@ -35,7 +35,7 @@ config_changelog = """
main = [
"llamafactory @ git+https://github.com/hiyouga/LLaMA-Factory.git",
"vllm==0.9.1; platform_system == 'Linux'",
"torch>=2.6.0",
"torch==2.7.0",
"transformers==4.52.1",
"accelerate==1.7.0",
"triton==3.2.0; platform_system == 'Linux'",
@@ -68,16 +68,16 @@ torch-backend = "auto"
[tool.uv.sources]
torch = [
{ index = "pytorch-cu124", marker = "platform_system == 'Windows'" },
{ index = "pytorch-cu124", marker = "platform_system == 'Linux'" },
{ index = "pytorch-cu126", marker = "platform_system == 'Windows'" },
{ index = "pytorch-cu126", marker = "platform_system == 'Linux'" },
]
torchaudio = [
{ index = "pytorch-cu124", marker = "platform_system == 'Windows'" },
{ index = "pytorch-cu124", marker = "platform_system == 'Linux'" },
{ index = "pytorch-cu126", marker = "platform_system == 'Windows'" },
{ index = "pytorch-cu126", marker = "platform_system == 'Linux'" },
]
torchvision = [
{ index = "pytorch-cu124", marker = "platform_system == 'Windows'" },
{ index = "pytorch-cu124", marker = "platform_system == 'Linux'" },
{ index = "pytorch-cu126", marker = "platform_system == 'Windows'" },
{ index = "pytorch-cu126", marker = "platform_system == 'Linux'" },
]
@@ -86,8 +86,8 @@ url = "https://pypi.tuna.tsinghua.edu.cn/simple/"
default = true
[[tool.uv.index]]
name = "pytorch-cu124"
url = "https://download.pytorch.org/whl/cu124"
name = "pytorch-cu126"
url = "https://download.pytorch.org/whl/cu126"
explicit = true
[tool.setuptools.packages.find]