Supported extensions: .pt or .safetensors