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 architectures;
20pub mod audio_processor;
21pub mod builder;
22pub mod common;
23pub mod definition;
24pub mod executor;
25pub mod gguf_config;
26pub mod gguf_engine_loader;
27pub mod gguf_runtime;
28pub mod hf_download;
29pub mod image_processor;
30pub mod loader;
31pub mod mel;
32pub mod models;
33pub mod moe;
34pub mod moe_config;
35pub mod registry;
36pub mod source;
37pub mod tensor_wrapper;
38pub mod tokenizer;
39pub mod weights;
40
41pub use architectures::{BertModelWrapper, ClipModelWrapper, WhisperModelWrapper};
42pub use builder::{DefaultModelBuilderFactory, SimpleModelBuilder};
43pub use common::{DecoderOnlyLLM, LlmRuntimeConfig};
44pub use definition::{ConfigManager, ModelDefinition};
45pub use executor::{
46    BertModelExecutor, ClipModelExecutor, LlmExecutor, StubModelExecutor, TtsModelExecutor,
47    WhisperModelExecutor,
48};
49pub use hf_download::HfDownloader;
50pub use image_processor::ClipImageProcessor;
51pub use loader::SafeTensorsLoader;
52pub use registry::{
53    Architecture, DefaultModelRegistry, ModelAlias, ModelDiscoveryEntry, ModelFormatType,
54};
55pub use source::{
56    DefaultModelSourceResolver, ModelFormat, ModelSourceConfig, ModelSourceResolver,
57    ResolvedModelSource,
58};
59pub use tensor_wrapper::CandleTensorWrapper;
60pub use tokenizer::{TokenizerFactory, TokenizerHandle};
61pub use weights::{default_weight_loader, StubWeightLoader, WeightLoaderHandle};
62
63pub use ferrum_interfaces::{ModelBuilder, ModelExecutor, WeightLoader};
64pub use ferrum_types::{
65    Activation, AttentionConfig, ModelConfig, ModelInfo, ModelType, NormType, Result, RopeScaling,
66};
ferrum_models/lib.rs

ferrum_models/
lib.rs
