bitsandbytes==0.45.0
datasets>=3.2.0
einops
loguru
mmengine==0.10.6
openpyxl
peft>=0.14.0
scikit-image
scipy
SentencePiece
tiktoken
torch
torchvision
transformers==4.48.0
transformers_stream_generator

[all]
bitsandbytes==0.45.0
datasets>=3.2.0
einops
loguru
mmengine==0.10.6
openpyxl
peft>=0.14.0
scikit-image
scipy
SentencePiece
tiktoken
torch
torchvision
transformers==4.48.0
transformers_stream_generator
deepspeed==0.16.2
mpi4py-mpich
modelscope

[deepspeed]
bitsandbytes==0.45.0
datasets>=3.2.0
einops
loguru
mmengine==0.10.6
openpyxl
peft>=0.14.0
scikit-image
scipy
SentencePiece
tiktoken
torch
torchvision
transformers==4.48.0
transformers_stream_generator
deepspeed==0.16.2
mpi4py-mpich

[modelscope]
bitsandbytes==0.45.0
datasets>=3.2.0
einops
loguru
mmengine==0.10.6
openpyxl
peft>=0.14.0
scikit-image
scipy
SentencePiece
tiktoken
torch
torchvision
transformers==4.48.0
transformers_stream_generator
modelscope
