torch==2.1.1
torchvision==0.16.1
accelerate==0.24.1
safetensors==0.4.0
transformers==4.35.0