Struct SafeTensorFile 

pub struct SafeTensorFile { /* private fields */ }

A parsed SafeTensors file backed by an in-memory byte buffer.

Implementations

impl SafeTensorFile

pub fn from_file(path: &Path) -> Result<Self, ModelError>

Parse a SafeTensors file from disk.

pub fn from_bytes(bytes: &[u8]) -> Result<Self, ModelError>

Parse a SafeTensors file from an in-memory byte slice.

pub fn tensor_names(&self) -> Vec<&str>

Returns a list of all tensor names in the file.

pub fn tensor_info(&self, name: &str) -> Option<TensorInfo>

Returns metadata for a tensor by name.

pub fn load_tensor(&self, name: &str) -> Result<Tensor, ModelError>

Load a single tensor by name, converting to F32 if necessary.

F16 and BF16 data are converted to F32. I32, I64, U8, and Bool are converted to F32 by casting each element.