From c75d80e7eae2baf74d85399df70e9f18d5a6fc4b Mon Sep 17 00:00:00 2001 From: Chuck Tang Date: Wed, 6 Nov 2024 17:02:42 -0800 Subject: [PATCH] commit change --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 6e82203b3a..34ec7695dd 100644 --- a/setup.py +++ b/setup.py @@ -57,7 +57,7 @@ 'accelerate>=0.25,<0.34', # 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.5.o,<2.5.2', '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',