bump torch version

mosaicml · Aug 23, 2023 · 8d5ee3c · 8d5ee3c
1 parent 78b796b
commit 8d5ee3c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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'] = [