Skip to content

Commit

Permalink
add cloud stores to foundry deps (#612)
Browse files Browse the repository at this point in the history
* add cloud stores to foundry deps

* also add boto explicitly
  • Loading branch information
dakinggg authored Sep 20, 2023
1 parent 10b221f commit 1a9c9c4
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -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',
Expand All @@ -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 = {}
Expand Down

0 comments on commit 1a9c9c4

Please sign in to comment.