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