From ca8bf111eabebfbfb4058d100ecea262ee5fb225 Mon Sep 17 00:00:00 2001 From: Mihir Patel Date: Wed, 6 Sep 2023 14:47:41 -0700 Subject: [PATCH] raise torch (#583) Co-authored-by: Daniel King <43149077+dakinggg@users.noreply.github.com> --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 5c4575748d..583a55fe97 100644 --- a/setup.py +++ b/setup.py @@ -51,7 +51,7 @@ 'accelerate>=0.20,<0.21', # for HF inference `device_map` 'transformers>=4.31,<4.32', 'mosaicml-streaming>=0.5.1,<0.6', - 'torch>=1.13.1,<=2.0.1', + 'torch>=1.13.1,<2.1.1', 'datasets==2.10.1', 'fsspec==2023.6.0', # newer version results in a bug in datasets that duplicates data 'sentencepiece==0.1.97',