Align torch requirement with CLI repo

Relax the upper bound of the torch requirement to align with the CLI repo so that we stop force-downgrading torch to 2.3.x. Signed-off-by: Ben Browning <[email protected]>
instructlab · Nov 6, 2024 · 7c5050a · 7c5050a
1 parent ac783f1
commit 7c5050a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -13,6 +13,6 @@ langchain-text-splitters
 openai>=1.13.3,<2.0.0
 tabulate>=0.9.0
 tenacity>=8.3.0,!=8.4.0
-torch>=2.3.0,<2.4.0
+torch>=2.3.0,<2.5.0
 transformers>=4.41.2
 xdg-base-dirs>=6.0.1