Skip to main content

ferrum_models/
lib.rs

1//! Ferrum 模型层
2//!
3//! 该 crate 负责围绕 `ferrum-interfaces`/`ferrum-types` 定义的核心抽象
4//! 提供模型定义解析、构建器与权重加载占位实现,确保上层可以在
5//! 重构阶段编译。
6
7// TTS / Whisper架构里存在大量在不同调用路径下"当下未使用"的字段/方法
8// (e.g. 推理期不走的 layers、只在特定 feature 下用的导入)。抑制到 crate 级,
9// 新写代码的真实警告由 engine / sampler / scheduler 等下游 crate 的 CI 门把守。
10#![allow(
11    dead_code,
12    unused_imports,
13    unused_variables,
14    unused_mut,
15    unused_parens,
16    unused_assignments
17)]
18
19pub mod audio_processor;
20pub mod common;
21pub mod definition;
22pub mod executor;
23pub mod gguf_config;
24pub mod gguf_engine_loader;
25pub mod gguf_runtime;
26pub mod hf_download;
27pub mod image_processor;
28pub mod loader;
29pub mod mel;
30pub mod models;
31pub mod moe;
32pub mod moe_config;
33pub mod multimodal;
34pub mod registry;
35pub mod source;
36pub mod tensor_wrapper;
37pub mod tokenizer;
38pub mod weight_format;
39
40pub use common::{DecoderOnlyLLM, LlmRuntimeConfig};
41pub use definition::{ConfigManager, ModelDefinition};
42pub use executor::{
43    BertModelExecutor, ClipModelExecutor, LlmExecutor, StubModelExecutor, TtsModelExecutor,
44    WhisperModelExecutor,
45};
46pub use hf_download::HfDownloader;
47pub use image_processor::ClipImageProcessor;
48pub use loader::SafeTensorsLoader;
49pub use multimodal::{BertModelWrapper, ClipModelWrapper, WhisperModelWrapper};
50pub use registry::{
51    Architecture, DefaultModelRegistry, ModelAlias, ModelDiscoveryEntry, ModelFormatType,
52};
53pub use source::{
54    DefaultModelSourceResolver, ModelFormat, ModelSourceConfig, ModelSourceResolver,
55    ResolvedModelSource,
56};
57pub use tensor_wrapper::CandleTensorWrapper;
58pub use tokenizer::{TokenizerFactory, TokenizerHandle};
59
60pub use ferrum_interfaces::ModelExecutor;
61pub use ferrum_types::{
62    Activation, AttentionConfig, ModelConfig, ModelInfo, ModelType, NormType, Result, RopeScaling,
63};