Update dependencies in pyproject.toml to specify versions for torch and torchvision, and add source configuration for UV tool
Browse files- pyproject.toml +15 -2
pyproject.toml
CHANGED
|
@@ -4,8 +4,8 @@ version = "0.1.0"
|
|
| 4 |
description = "Local smoke-test environment for ViL-DLM"
|
| 5 |
requires-python = ">=3.11,<3.12"
|
| 6 |
dependencies = [
|
| 7 |
-
"torch",
|
| 8 |
-
"torchvision",
|
| 9 |
"transformers",
|
| 10 |
"huggingface_hub",
|
| 11 |
"einops",
|
|
@@ -22,3 +22,16 @@ dev = [
|
|
| 22 |
|
| 23 |
[tool.uv]
|
| 24 |
package = false
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
description = "Local smoke-test environment for ViL-DLM"
|
| 5 |
requires-python = ">=3.11,<3.12"
|
| 6 |
dependencies = [
|
| 7 |
+
"torch==2.7.1",
|
| 8 |
+
"torchvision==0.22.1",
|
| 9 |
"transformers",
|
| 10 |
"huggingface_hub",
|
| 11 |
"einops",
|
|
|
|
| 22 |
|
| 23 |
[tool.uv]
|
| 24 |
package = false
|
| 25 |
+
|
| 26 |
+
[tool.uv.sources]
|
| 27 |
+
torch = [
|
| 28 |
+
{ index = "pytorch-cu128", marker = "platform_system == 'Linux'" },
|
| 29 |
+
]
|
| 30 |
+
torchvision = [
|
| 31 |
+
{ index = "pytorch-cu128", marker = "platform_system == 'Linux'" },
|
| 32 |
+
]
|
| 33 |
+
|
| 34 |
+
[[tool.uv.index]]
|
| 35 |
+
name = "pytorch-cu128"
|
| 36 |
+
url = "https://download.pytorch.org/whl/cu128"
|
| 37 |
+
explicit = true
|