torch
torchvision
safetensors