Struct ShardIndex 

pub struct ShardIndex { /* private fields */ }

Parsed shard index from HuggingFace format.

Represents the model.safetensors.index.json file structure:

{
  "metadata": {...},
  "weight_map": {
    "layer.0.weight": "model-00001-of-00006.safetensors",
    "layer.1.weight": "model-00002-of-00006.safetensors"
  }
}

Implementations

impl ShardIndex

pub fn new() -> Self

Create empty shard index

pub fn from_json(json: &str) -> Result<Self>

Parse from HuggingFace index.json format.

Errors

Returns error if JSON parsing fails.

pub fn add_mapping(&mut self, tensor_name: &str, shard_file: &str)

Add tensor to shard mapping

pub fn shard_for_tensor(&self, tensor_name: &str) -> Option<&str>

Get shard file for a tensor

pub fn shard_index(&self, shard_file: &str) -> Option<usize>

Get shard index (0-based) for a shard file

pub fn shard_count(&self) -> usize

Get total number of shards

pub fn tensor_count(&self) -> usize

Get total number of tensors

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

Get all shard filenames in order

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

Get all tensor names (sorted for consistent processing)

pub fn tensors_by_shard(&self) -> HashMap<&str, Vec<&str>>

Group tensors by shard for efficient loading

pub fn is_valid(&self) -> bool

Check if index is valid

pub fn set_metadata(&mut self, key: &str, value: &str)

Set metadata

pub fn get_metadata(&self, key: &str) -> Option<&str>

Get metadata

Trait Implementations

impl Clone for ShardIndex

fn clone(&self) -> ShardIndex

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for ShardIndex

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

Formats the value using the given formatter.

impl Default for ShardIndex

fn default() -> ShardIndex

Returns the “default value” for a type.