torch numpy Image torchvision transformers==4.24.0