From 55166d3227c91554653397068cba13c02589ea4d Mon Sep 17 00:00:00 2001
From: Lorenzo Stella <stellalo@amazon.com>
Date: Sat, 18 May 2024 06:53:40 +0200
Subject: [PATCH] Cap transformers <4.41 (#77)

*Issue #, if available:* #76

*Description of changes:* transformers 4.41 broke something for us, we
need to look into it deeper


By submitting this pull request, I confirm that you can use, modify,
copy, and redistribute this contribution, under the terms of your
choice.
---
 pyproject.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 28c096b..14f3ec4 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -5,7 +5,7 @@ requires-python = ">=3.8"
 license = { file = "LICENSE" }
 dependencies = [
     "torch~=2.1",         # package was tested on 2.2
-    "transformers~=4.31",
+    "transformers~=4.31,<4.41",
     "accelerate",
 ]