ferrum_models/
lib.rs

1//! Ferrum 模型层
2//!
3//! 该 crate 负责围绕 `ferrum-interfaces`/`ferrum-types` 定义的核心抽象
4//! 提供模型定义解析、构建器与权重加载占位实现，确保上层可以在
5//! 重构阶段编译。
6
7pub mod architectures;
8pub mod builder;
9pub mod definition;
10pub mod executor;
11pub mod hf_download;
12pub mod image_processor;
13pub mod loader;
14pub mod registry;
15pub mod source;
16pub mod tensor_wrapper;
17pub mod tokenizer;
18pub mod weights;
19
20pub use architectures::{
21    BertModelWrapper, ClipModelWrapper, LlamaModelWrapper, Qwen2ModelWrapper, Qwen3ModelWrapper,
22};
23pub use builder::{DefaultModelBuilderFactory, SimpleModelBuilder};
24pub use definition::{ConfigManager, ModelDefinition};
25pub use executor::{
26    BertModelExecutor, CandleModelExecutor, ClipModelExecutor, Qwen2ModelExecutor,
27    Qwen3ModelExecutor, StubModelExecutor,
28};
29pub use hf_download::HfDownloader;
30pub use image_processor::ClipImageProcessor;
31pub use loader::SafeTensorsLoader;
32pub use registry::{
33    Architecture, DefaultModelRegistry, ModelAlias, ModelDiscoveryEntry, ModelFormatType,
34};
35pub use source::{
36    DefaultModelSourceResolver, ModelFormat, ModelSourceConfig, ModelSourceResolver,
37    ResolvedModelSource,
38};
39pub use tensor_wrapper::CandleTensorWrapper;
40pub use tokenizer::{TokenizerFactory, TokenizerHandle};
41pub use weights::{default_weight_loader, StubWeightLoader, WeightLoaderHandle};
42
43pub use ferrum_interfaces::{ModelBuilder, ModelExecutor, WeightLoader};
44pub use ferrum_types::{
45    Activation, AttentionConfig, ModelConfig, ModelInfo, ModelType, NormType, Result, RopeScaling,
46};
ferrum_models/lib.rs

ferrum_models/
lib.rs
