update torch dep in framework

Signed-off-by: Yu Chin Fabian Lim <flim@sg.ibm.com>
foundation-model-stack · May 17, 2024 · b93c8c3 · b93c8c3
1 parent 0aa4437
commit b93c8c3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/plugins/framework/pyproject.toml b/plugins/framework/pyproject.toml
@@ -22,7 +22,7 @@ classifiers=[
     "Programming Language :: Python :: 3.11",
 ]
 dependencies = [
-  "torch<2.3",
+  "torch>2.2,<2.3",
   "transformers<4.40",
   "peft",
   "accelerate"