mosaicml[gcs,libcloud,mlflow,oci,wandb]<0.17,>=0.16.3
accelerate<0.21,>=0.20
transformers<4.34,>=4.33
mosaicml-streaming<0.7,>=0.6
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

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

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

[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

[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.17,>=0.16.1
