build(deps): loosen torch req to <2.3

amitkparekh · Aug 1, 2024 · fb1c5b1 · fb1c5b1
1 parent dcc659c
commit fb1c5b1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -9,7 +9,7 @@ requires-python = ">=3.11,<3.13"
 readme = "README.md"
 license = {text = "MIT"}
 dependencies = [
-    "torch==2.2.0",
+    "torch>=2.2.0,<2.3",
     "tokenizers>=0.13.3",
     "transformers>=4.32.0",
     "omegaconf>=2.3.0",