ferrum_models/
lib.rs

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

ferrum_models/
lib.rs
