We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ba6e43c commit a1fce9eCopy full SHA for a1fce9e
setup.py
@@ -54,7 +54,7 @@
54
install_requires = [
55
'mosaicml[libcloud,wandb,oci,gcs,mlflow]>=0.30.0,<0.31',
56
'mlflow>=2.14.1,<2.22',
57
- 'accelerate>=0.25,<1.4', # for HF inference `device_map`
+ 'accelerate>=0.25,<1.7', # for HF inference `device_map`
58
'transformers>=v4.51.0,<4.52',
59
'mosaicml-streaming>=0.12.0,<0.13',
60
'torch>=2.6.0,<2.6.1',
0 commit comments