From 8356e1f53bd309c6f22935a0a27fa20bac143cde Mon Sep 17 00:00:00 2001
From: Chuck Tang <chuck.tang@databricks.com>
Date: Wed, 20 Nov 2024 14:24:53 -0800
Subject: [PATCH] commit change

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index b401f4d7b8..566e6aae9c 100644
--- a/setup.py
+++ b/setup.py
@@ -57,7 +57,7 @@
     'accelerate>=0.25,<1.2',  # for HF inference `device_map`
     'transformers>=4.43.2,<4.47',
     'mosaicml-streaming>=0.9.0,<0.10',
-    'torch>=2.5.1,<2.5.2',
+    'torch>=2.4.0,<2.4.1',
     'datasets>=2.20.0,<2.21',
     'fsspec==2023.6.0',  # newer version results in a bug in datasets that duplicates data
     'sentencepiece==0.2.0',