From c9a66d920caf1cd9b628b044826208027babcdbc Mon Sep 17 00:00:00 2001 From: Daniel King Date: Tue, 19 Sep 2023 18:06:55 -0700 Subject: [PATCH] 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',