Skip to content

Commit c0b360f

Browse files
committed
update accelerate based on auto-round requirement
Signed-off-by: Dan Huang <[email protected]>
1 parent d59b7e9 commit c0b360f

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

setup.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -128,8 +128,9 @@ def localversion_func(version: ScmVersion) -> str:
128128
else "transformers>=4.54.0"
129129
),
130130
("datasets>=4.0.0,<=4.4.1" if BUILD_TYPE == "release" else "datasets>=4.0.0"),
131+
# auto-round 0.9.1 requires accelerate >= 1.10.0
131132
(
132-
"accelerate>=1.6.0,<=1.12.0"
133+
"accelerate>=1.10.0,<=1.12.0"
133134
if BUILD_TYPE == "release"
134135
else "accelerate>=1.6.0"
135136
),

0 commit comments

Comments
 (0)