transformers
protobuf
python-socketio[client]
tokenizers>=0.13.0
pydantic>=2.4.0
torch>=2.1.0
sentencepiece
torchvision
accelerate
diffusers
einops
