Skip to content

Commit 2480ab5

Browse files
authored
Testing removing nvidia drivers (#1277)
1 parent 533b91f commit 2480ab5

File tree

2 files changed

+66
-182
lines changed

2 files changed

+66
-182
lines changed

pyproject.toml

Lines changed: 13 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,3 @@
1-
[tool.uv.pip]
2-
extra-index-url = ["https://download.pytorch.org/whl/cpu"]
31

42
[project]
53
name = "merino-py"
@@ -33,11 +31,22 @@ dependencies = [
3331
"ua-parser>=1.0,<2.0",
3432
"redis[hiredis]>=5.2.1,<6",
3533
"circuitbreaker>=2.1.3",
36-
"torch>=2.9.1,<3",
37-
"safetensors>=0.7.0,<1",
3834
"mozilla-merino-ext>=0.2.0",
35+
"torch>=2.9.1",
36+
"safetensors>=0.7.0,<1"
3937
]
4038

39+
[[tool.uv.index]]
40+
name = "pytorch-cpu"
41+
url = "https://download.pytorch.org/whl/cpu"
42+
explicit = true
43+
44+
[tool.uv.sources]
45+
torch = [
46+
{ index = "pytorch-cpu" },
47+
]
48+
49+
4150
[project.scripts]
4251
merino-jobs = "merino.jobs.cli:cli"
4352
probe-images = "merino.jobs.utils.domain_tester:main"

0 commit comments

Comments
 (0)