Skip to content

Commit 3ec6d5c

Browse files
committed
add torch 2.6
1 parent cd7be3d commit 3ec6d5c

File tree

2 files changed

+70
-28
lines changed

2 files changed

+70
-28
lines changed

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ description = "ZeroBand is a production ready codebase for decentralized trainin
55
readme = "README.md"
66
requires-python = ">=3.10"
77
dependencies = [
8-
"torch==2.5.1",
8+
"torch==2.6.0",
99
"numpy",
1010
"setuptools",
1111
"transformers>=4.44.2",

0 commit comments

Comments
 (0)