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 loader;
13pub mod registry;
14pub mod source;
15pub mod tensor_wrapper;
16pub mod tokenizer;
17pub mod weights;
18
19pub use architectures::{
20    BertModelWrapper, LlamaModelWrapper, Qwen2ModelWrapper, Qwen3ModelWrapper,
21};
22pub use builder::{DefaultModelBuilderFactory, SimpleModelBuilder};
23pub use definition::{ConfigManager, ModelDefinition};
24pub use executor::{
25    BertModelExecutor, CandleModelExecutor, Qwen2ModelExecutor, Qwen3ModelExecutor,
26    StubModelExecutor,
27};
28pub use hf_download::HfDownloader;
29pub use loader::SafeTensorsLoader;
30pub use registry::{
31    Architecture, DefaultModelRegistry, ModelAlias, ModelDiscoveryEntry, ModelFormatType,
32};
33pub use source::{
34    DefaultModelSourceResolver, ModelFormat, ModelSourceConfig, ModelSourceResolver,
35    ResolvedModelSource,
36};
37pub use tensor_wrapper::CandleTensorWrapper;
38pub use tokenizer::{TokenizerFactory, TokenizerHandle};
39pub use weights::{default_weight_loader, StubWeightLoader, WeightLoaderHandle};
40
41pub use ferrum_interfaces::{ModelBuilder, ModelExecutor, WeightLoader};
42pub use ferrum_types::{
43    Activation, AttentionConfig, ModelConfig, ModelInfo, ModelType, NormType, Result, RopeScaling,
44};
ferrum_models/lib.rs

ferrum_models/
lib.rs
