Skip to content

Commit d67d90f

Browse files
committed
Set torch upper bound
1 parent 3b6682c commit d67d90f

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

sdks/python/setup.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -198,7 +198,8 @@ def cythonize(*args, **kwargs):
198198
# Drop this cap once transformers updates the CLIP call site to
199199
# `cls_token=` or tokenizers reinstates `cls=` as a deprecation alias.
200200
'tokenizers<0.23',
201-
'torch',
201+
# Avoid torch 2.12.0+ which fails to run unit tests with segfault.
202+
'torch<2.12.0',
202203
# Match tested transformers range.
203204
'transformers>=4.28.0,<4.56.0',
204205
# Keep tokenizers compatible with this transformers range.

0 commit comments

Comments
 (0)