fix

Author: Abdul Fatir Ansari

pyproject.toml

@@ -4,7 +4,7 @@ version = "1.3.0"
 requires-python = ">=3.8"
 license = { file = "LICENSE" }
 dependencies = [
-    "torch>=2.0,2.6",         # package was tested on 2.2
+    "torch>=2.0,<2.6",         # package was tested on 2.2
     "transformers>=4.30,<4.48",
     "accelerate>=0.32,<1",
 ]