bitsandbytes>=0.40.0
datasets
einops
lagent>=0.1.2
mmengine>=0.9.0
modelscope
peft>=0.4.0
scipy
SentencePiece
tiktoken
torch
transformers>=4.32.1
transformers_stream_generator

[all]
bitsandbytes>=0.40.0
datasets
einops
lagent>=0.1.2
mmengine>=0.9.0
modelscope
peft>=0.4.0
scipy
SentencePiece
tiktoken
torch
transformers>=4.32.1
transformers_stream_generator
deepspeed
mpi4py-mpich

[deepspeed]
bitsandbytes>=0.40.0
datasets
einops
lagent>=0.1.2
mmengine>=0.9.0
modelscope
peft>=0.4.0
scipy
SentencePiece
tiktoken
torch
transformers>=4.32.1
transformers_stream_generator
deepspeed
mpi4py-mpich
