torch==2.1.2
transformers==4.36.2
torchvision==0.16.2
einops==0.7.0
safetensors==0.4.1