From 8d5ee3ca19612a4ad982d25cb094e49aa4d48949 Mon Sep 17 00:00:00 2001
From: Chuck Tang <chuck@mosaicml.com>
Date: Wed, 23 Aug 2023 16:57:17 -0700
Subject: [PATCH] bump torch version

---
 setup.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/setup.py b/setup.py
index f32876a943..d61b7ce2a5 100644
--- a/setup.py
+++ b/setup.py
@@ -47,11 +47,11 @@
 ]
 
 install_requires = [
-    'mosaicml[libcloud,wandb,mlflow]>=0.15.0,<0.16',
+    'mosaicml[libcloud,wandb,mlflow]>=0.15.0,<0.16.2',
     '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.0',
     'datasets==2.10.1',
     'sentencepiece==0.1.97',
     'einops==0.5.0',
@@ -79,7 +79,7 @@
 ]
 
 extra_deps['tensorboard'] = [
-    'mosaicml[tensorboard]>=0.15.0,<0.16',
+    'mosaicml[tensorboard]>=0.15.0,<0.16.2',
 ]
 
 extra_deps['gpu'] = [