mosaicml[gcs,libcloud,mlflow,oci,wandb]<0.18,>=0.17
accelerate<0.21,>=0.20
transformers<4.35,>=4.34.1
mosaicml-streaming<0.8,>=0.7.1
torch<2.1.1,>=1.13.1
datasets<2.15,>=2.14.5
fsspec==2023.6.0
sentencepiece==0.1.97
einops==0.5.0
omegaconf<3,>=2.2.3
slack-sdk<4
mosaicml-cli<1,>=0.3
onnx==1.14.0
onnxruntime==1.15.1
cmake<=3.26.3,>=3.25.0
boto3<2,>=1.21.45
huggingface-hub<1.0,>=0.17.0
beautifulsoup4<5,>=4.12.2
tenacity<9,>=8.2.3

[all]
mosaicml-turbo==0.0.4
packaging<23,>=21
pytest<8,>=7.2.1
mosaicml[tensorboard]<0.18,>=0.17
bitsandbytes==0.39.1
hf_transfer==0.1.3
toml<0.11,>=0.10.2
openai==0.27.8
pytest_codeblocks<0.17,>=0.16.1
pytest-cov<5,>=4
peft==0.4.0
tiktoken==0.4.0
pre-commit<3,>=2.18.1
flash-attn==1.0.9
mosaicml[databricks]<0.18,>=0.17
loralib==0.1.1
scipy<=1.11.0,>=1.10.0
pyright==1.1.256

[all-cpu]
packaging<23,>=21
pytest<8,>=7.2.1
mosaicml[tensorboard]<0.18,>=0.17
bitsandbytes==0.39.1
hf_transfer==0.1.3
toml<0.11,>=0.10.2
openai==0.27.8
pytest_codeblocks<0.17,>=0.16.1
pytest-cov<5,>=4
peft==0.4.0
tiktoken==0.4.0
pre-commit<3,>=2.18.1
scipy<=1.11.0,>=1.10.0
mosaicml[databricks]<0.18,>=0.17
loralib==0.1.1
pyright==1.1.256

[all-flash2]
mosaicml-turbo==0.0.4
packaging<23,>=21
pytest<8,>=7.2.1
mosaicml[tensorboard]<0.18,>=0.17
bitsandbytes==0.39.1
hf_transfer==0.1.3
toml<0.11,>=0.10.2
openai==0.27.8
pytest_codeblocks<0.17,>=0.16.1
pytest-cov<5,>=4
flash-attn==2.3.2
peft==0.4.0
tiktoken==0.4.0
pre-commit<3,>=2.18.1
scipy<=1.11.0,>=1.10.0
mosaicml[databricks]<0.18,>=0.17
loralib==0.1.1
pyright==1.1.256

[databricks]
mosaicml[databricks]<0.18,>=0.17

[dev]
pre-commit<3,>=2.18.1
pytest<8,>=7.2.1
pytest_codeblocks<0.17,>=0.16.1
pytest-cov<5,>=4
pyright==1.1.256
toml<0.11,>=0.10.2
packaging<23,>=21
hf_transfer==0.1.3

[gpu]
flash-attn==1.0.9
mosaicml-turbo==0.0.4

[gpu-flash2]
flash-attn==2.3.2
mosaicml-turbo==0.0.4

[openai]
openai==0.27.8
tiktoken==0.4.0

[peft]
loralib==0.1.1
bitsandbytes==0.39.1
scipy<=1.11.0,>=1.10.0
peft==0.4.0

[tensorboard]
mosaicml[tensorboard]<0.18,>=0.17
