torch
safetensors