Struct HfSafetensors 

pub struct HfSafetensors { /* private fields */ }

HuggingFace SafeTensors model loader

Implementations

impl HfSafetensors

pub fn from_file(path: &Path) -> Result<HfSafetensors, SafetensorsError>

Load from a local safetensors file

pub fn from_bytes(data: Vec<u8>) -> Result<HfSafetensors, SafetensorsError>

Load from raw bytes

pub fn from_hub(repo_id: &str) -> Result<HfSafetensors, SafetensorsError>

Download from HuggingFace Hub

pub fn tensor_names(&self) -> &[String]

List all tensor names

pub fn tensor(&self, name: &str) -> Result<TensorData, SafetensorsError>

Get a specific tensor by name

pub fn compare_all<F>( &self, get_apr_tensor: F, threshold: f64, ) -> Vec<TensorComparison>

where F: Fn(&str) -> Option<Vec<f32>>,

Compare all tensors against an APR model

Trait Implementations

impl Debug for HfSafetensors

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

Formats the value using the given formatter.