Updates torch to use the most recent version (#822)

* removed torch ceiling * updated torch version
TransformerLensOrg · Dec 27, 2024 · 2bde039 · 2bde039
1 parent 7aed579
commit 2bde039
Show file tree

Hide file tree

Showing 2 changed files with 49 additions and 47 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -28,10 +28,7 @@
         python=">=3.9,<4.0"
         rich=">=12.6.0"
         sentencepiece="*"
-        torch=[
-            {platform="!=linux", version=">=2.0,!=2.0,!=2.1.0,<2.2"}, # Pin >=2.1.1 on Apple devices due to known MPS errors on 2.1.0
-            {platform="linux", version=">=2.0,<2.2"},                 # We can use any torch version on Linux (e.g colab)
-        ]
+        torch=">=2.2,<2.5"
         tqdm=">=4.64.1"
         transformers=">=4.37.2,<4.43"
         typing-extensions="*"