Skip to content

Commit

Permalink
Bump transformers and stop at <= 4.44.0
Browse files Browse the repository at this point in the history
* Use >= 4.38.0 because that is the sentence-transformers min anyway.
* Add <= 4.44.0 because 4.44.0 breaks our tests with errors like this:
  > `RuntimeError: The expanded size of the tensor (22) must match the existing size (30) at non-singleton dimension 2.  Target sizes: [4, 22, 22].  Tensor sizes: [4, 1, 30]`

Signed-off-by: Mark Sturdevant <mark.sturdevant@ibm.com>
  • Loading branch information
markstur committed Aug 12, 2024
1 parent 3343d88 commit bcc1233
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ dependencies = [
"tokenizers>=0.13.3",
"torch>=2.3.1,<2.4.0",
"tqdm>=4.65.0",
"transformers>=4.32.0",
"transformers>=4.38.0,<4.44.0",
"peft==0.6.0",
]

Expand Down

0 comments on commit bcc1233

Please sign in to comment.