From c9a66d920caf1cd9b628b044826208027babcdbc Mon Sep 17 00:00:00 2001 From: Daniel King Date: Tue, 19 Sep 2023 18:06:55 -0700 Subject: [PATCH 1/2] add cloud stores to foundry deps --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 5cd1309922..6d9537c210 100644 --- a/setup.py +++ b/setup.py @@ -47,7 +47,7 @@ ] install_requires = [ - 'mosaicml[libcloud,wandb,mlflow]>=0.16.1,<0.17', + 'mosaicml[libcloud,wandb,mlflow,oci,gcs]>=0.16.1,<0.17', 'accelerate>=0.20,<0.21', # for HF inference `device_map` 'transformers>=4.33,<4.34', 'mosaicml-streaming>=0.6,<0.7', From 6fd6bb75b7228529fa9a257d417f67fc226aeda4 Mon Sep 17 00:00:00 2001 From: Daniel King Date: Tue, 19 Sep 2023 18:11:25 -0700 Subject: [PATCH 2/2] also add boto explicitly --- setup.py | 1 + 1 file changed, 1 insertion(+) diff --git a/setup.py b/setup.py index 6d9537c210..f86b03bbab 100644 --- a/setup.py +++ b/setup.py @@ -64,6 +64,7 @@ 'cmake>=3.25.0,<=3.26.3', # required for triton-pre-mlir below # PyPI does not support direct dependencies, so we remove this line before uploading from PyPI 'triton-pre-mlir@git+https://github.com/vchiley/triton.git@triton_pre_mlir_sm90#subdirectory=python', + 'boto3>=1.21.45,<2', ] extra_deps = {}