Struct SafetensorsFile 

pub struct SafetensorsFile { /* private fields */ }

A memory-mapped safetensors file ready for tensor extraction.

This struct owns the mmap and parsed header. Individual tensors can be loaded into Metal buffers on demand via load_tensor or all at once via load_all_tensors.

Implementations

impl SafetensorsFile

pub fn open(path: &Path) -> Result<Self>

Open and memory-map a safetensors file.

The file is mapped read-only. No tensor data is copied until load_tensor or load_all_tensors is called.

Errors

• MlxError::IoError if the file cannot be opened or mapped.

pub fn tensor_names(&self) -> Result<Vec<String>>

List all tensor names in the file.

pub fn load_tensor( &self, name: &str, device: &MlxDevice, ) -> Result<(DType, Vec<usize>, MlxBuffer)>

Load a single named tensor into a Metal buffer.

Returns the dtype, shape, and a Metal buffer containing the raw bytes.

Errors

• MlxError::SafetensorsError if the tensor name is not found.
• MlxError::UnsupportedDtype if the tensor’s dtype is not supported.
• MlxError::BufferAllocationError if Metal allocation fails.

pub fn load_all_tensors( &self, device: &MlxDevice, ) -> Result<HashMap<String, (DType, Vec<usize>, MlxBuffer)>>

Load all tensors from the file into Metal buffers.

Returns a map from tensor name to (DType, shape, MlxBuffer).

Errors

Returns the first error encountered during loading.

Trait Implementations

impl Debug for SafetensorsFile

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.