Skip to content

Commit f8dd176

Browse files
committed
fix issues with dependencies for bert scorer test
1 parent cd28dcd commit f8dd176

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

pyproject.toml

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,9 @@ dependencies = [
5151
[project.optional-dependencies]
5252
catboost = ["catboost (>=1.2.8,<2.0.0)"]
5353
peft = ["peft (>= 0.10.0, !=0.15.0, !=0.15.1, <1.0.0)"]
54-
transformers = ["transformers (>=4.49.0,<5.0.0)"]
54+
transformers = [
55+
"transformers[torch] (>=4.49.0,<5.0.0)",
56+
]
5557
sentence-transformers = ["sentence-transformers (>=3,<4)"]
5658
dspy = [
5759
"dspy (>=2.6.5,<3.0.0)",

0 commit comments

Comments
 (0)