torch==2.1.0
torchvision==0.15.2
accelerate==0.24.1
safetensors==0.4.0
transformers==4.35.0