Skip to main content

Module loader

Module loader 

Source
Expand description

Model loader for loading pre-trained weights

This module provides:

  • Safetensors format parsing
  • Model configuration (LLaMA, Mistral)
  • Weight mapping to model structures
  • HuggingFace Hub integration (optional)

Re-exports§

pub use safetensors::SafetensorsLoader;
pub use config::ModelConfig;
pub use config::LlamaConfig;
pub use config::MistralConfig;
pub use weight_mapper::LlamaWeightMapper;

Modules§

config
Model configuration for LLaMA and Mistral
safetensors
Safetensors format loader
weight_mapper
Weight mapper for converting Safetensors weights to model structures