Skip to content

Commit

Permalink
Lock pytorch
Browse files Browse the repository at this point in the history
  • Loading branch information
druzsan committed Nov 8, 2024
1 parent 27097fa commit 294c727
Show file tree
Hide file tree
Showing 2 changed files with 142 additions and 13 deletions.
150 changes: 139 additions & 11 deletions poetry.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

5 changes: 3 additions & 2 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -127,14 +127,15 @@ types-pillow = "^10.0.0.1"
pandas-stubs = "^2.0.2.230605"
ruff = "^0.2.1"
check-wheel-contents = "^0.6.0"
torch = { version = "^2.1.1+cpu", source = "torch-cpu" }
torch = { version = "^2.1.1+cpu, !=2.5.1+cpu", source = "torch-cpu" }
ray = {extras = ["data"], version = "==2.9.3"}
soxr = "<0.5.0"

[tool.poetry.group.playbook.dependencies]
towhee = "^0.9.0"
annoy = "^1.17.2"
cleanlab = "^2.4.0"
torch = { version = "^2.1.1+cpu", source = "torch-cpu" }
torch = { version = "^2.1.1+cpu, !=2.5.1+cpu", source = "torch-cpu" }

[[tool.poetry.source]]
name = "torch-cpu"
Expand Down

0 comments on commit 294c727

Please sign in to comment.