Use PyTorch cu132 nightly for Windows
Browse files- pyproject.toml +4 -4
- uv.lock +0 -0
pyproject.toml
CHANGED
|
@@ -14,7 +14,7 @@ dependencies = [
|
|
| 14 |
"onnxscript==0.7.0",
|
| 15 |
"seqeval==1.2.2",
|
| 16 |
"tensorboard>=2.14.0",
|
| 17 |
-
"torch==2.
|
| 18 |
"transformers==5.8.1",
|
| 19 |
]
|
| 20 |
|
|
@@ -27,10 +27,10 @@ environments = ["sys_platform == 'win32'"]
|
|
| 27 |
|
| 28 |
[tool.uv.sources]
|
| 29 |
torch = [
|
| 30 |
-
{ index = "pytorch-
|
| 31 |
]
|
| 32 |
|
| 33 |
[[tool.uv.index]]
|
| 34 |
-
name = "pytorch-
|
| 35 |
-
url = "https://download.pytorch.org/whl/
|
| 36 |
explicit = true
|
|
|
|
| 14 |
"onnxscript==0.7.0",
|
| 15 |
"seqeval==1.2.2",
|
| 16 |
"tensorboard>=2.14.0",
|
| 17 |
+
"torch==2.13.0.dev20260601+cu132",
|
| 18 |
"transformers==5.8.1",
|
| 19 |
]
|
| 20 |
|
|
|
|
| 27 |
|
| 28 |
[tool.uv.sources]
|
| 29 |
torch = [
|
| 30 |
+
{ index = "pytorch-cu132-nightly", marker = "platform_system == 'Windows'" },
|
| 31 |
]
|
| 32 |
|
| 33 |
[[tool.uv.index]]
|
| 34 |
+
name = "pytorch-cu132-nightly"
|
| 35 |
+
url = "https://download.pytorch.org/whl/nightly/cu132"
|
| 36 |
explicit = true
|
uv.lock
CHANGED
|
The diff for this file is too large to render.
See raw diff
|
|
|