Skip to content

Commit 98eac0c

Browse files
authored
Fix: Transformers lib compatibility (#110)
Closes #109
1 parent 1427634 commit 98eac0c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pyproject.toml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ version = "3.1.2"
4747
uform-chat = "uform.chat:main"
4848

4949
[project.optional-dependencies]
50-
torch = ["torch>=1.13.1", "torchvision", "transformers>=4.36.2"]
50+
torch = ["torch>=1.13.1", "torchvision", "transformers>=4.36.2,<4.49.0"]
5151
onnx = ["onnx>=1.15.0", "onnxruntime>=1.17.1", "numpy"]
5252
onnx-gpu = ["onnx>=1.15.0", "onnxruntime-gpu>=1.17.1", "numpy"]
5353
dev = ["pytest", "pandas"]
@@ -85,4 +85,4 @@ trailing_comma_inline_array = true
8585
# https://black.readthedocs.io/en/latest/usage_and_configuration/the_basics.html#where-black-looks-for-the-file
8686
[tool.black]
8787
line-length = 120 # Set line length to the same value as in `.clang-format` for modern wide screens
88-
target-version = ['py36', 'py312'] # Set target Python versions to 3.6 and 3.12
88+
target-version = ['py36', 'py312'] # Set target Python versions to 3.6 and 3.12

0 commit comments

Comments
 (0)