Skip to content

Commit 0f7e998

Browse files
committed
fix: bump up transformers for tp_size feature
Signed-off-by: Mehant Kammakomati <mehant.kammakomati2@ibm.com>
1 parent 02b98cd commit 0f7e998

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/accelerate/utils/constants.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@
4949
XPU_PROFILING_AVAILABLE_PYTORCH_VERSION = "2.4.0"
5050
MITA_PROFILING_AVAILABLE_PYTORCH_VERSION = "2.1.0"
5151
BETA_TP_AVAILABLE_PYTORCH_VERSION = "2.3.0"
52-
BETA_TP_AVAILABLE_TRANSFORMERS_VERSION = "4.50.0"
52+
BETA_TP_AVAILABLE_TRANSFORMERS_VERSION = "4.52.0"
5353

5454
STR_OPERATION_TO_FUNC = {">": op.gt, ">=": op.ge, "==": op.eq, "!=": op.ne, "<=": op.le, "<": op.lt}
5555

0 commit comments

Comments
 (0)