torch 
transformers==4.31.0
torchvision