torch transformers pyyaml numpy safetensors