einops==0.7.0
protobuf==5.26.1
torch==2.5.0
torchvision==0.20.0
torchaudio==2.5.0 --index-url https://download.pytorch.org/whl/cu121
transformers==4.39.0