god-graph 0.6.0-alpha

A graph-based LLM white-box optimization toolbox: topology validation, Lie group orthogonalization, tensor ring compression
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
//! 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)

pub mod safetensors;
pub mod config;
pub mod weight_mapper;

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