From bcc1233221862280258c8d646ba15c08bdcd1428 Mon Sep 17 00:00:00 2001 From: Mark Sturdevant Date: Mon, 12 Aug 2024 11:35:01 -0700 Subject: [PATCH] Bump transformers and stop at <= 4.44.0 * 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 --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index cbf6e401..3a2464f7 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -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", ]