Skip to content

Commit

Permalink
Update composer to 0.17.2 and streaming to 0.7.2 (#822)
Browse files Browse the repository at this point in the history
  • Loading branch information
irenedea authored Jan 3, 2024
1 parent 29b41ba commit 1c4ebbf
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -47,10 +47,10 @@
]

install_requires = [
'mosaicml[libcloud,wandb,mlflow,oci,gcs]>=0.17.1,<0.18',
'mosaicml[libcloud,wandb,mlflow,oci,gcs]>=0.17.2,<0.18',
'accelerate>=0.25,<0.26', # for HF inference `device_map`
'transformers>=4.36,<4.37',
'mosaicml-streaming>=0.7.1,<0.8',
'mosaicml-streaming>=0.7.2,<0.8',
'torch>=2.1,<2.1.1',
'datasets==2.15.0',
'fsspec==2023.6.0', # newer version results in a bug in datasets that duplicates data
Expand Down Expand Up @@ -84,11 +84,11 @@
]

extra_deps['databricks'] = [
'mosaicml[databricks]>=0.17.1,<0.18',
'mosaicml[databricks]>=0.17.2,<0.18',
]

extra_deps['tensorboard'] = [
'mosaicml[tensorboard]>=0.17.1,<0.18',
'mosaicml[tensorboard]>=0.17.2,<0.18',
]

extra_deps['gpu'] = [
Expand Down

0 comments on commit 1c4ebbf

Please sign in to comment.