raise torch (#583)

Co-authored-by: Daniel King <[email protected]>

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',