Struct SafetensorFile 

pub struct SafetensorFile { /* private fields */ }

Implementations

impl SafetensorFile

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

pub fn path(&self) -> &Path

pub fn data_start(&self) -> u64

pub fn tensors(&self) -> &BTreeMap<String, SafetensorTensorInfo>

pub fn stream_tensor<W: Write>( &self, name: &str, writer: &mut W, buffer_size: usize, ) -> Result<u64>

pub fn stream_tensor_chunks<F>( &self, name: &str, buffer_size: usize, on_chunk: F, ) -> Result<u64>

where F: FnMut(&[u8]) -> Result<()>,

Trait Implementations

impl Debug for SafetensorFile

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

Formats the value using the given formatter.