1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
//! Models V2 - Clean implementations using solid abstractions
//!
//! This module contains model implementations that use our new solid abstractions:
//! - TensorCore for unified tensor operations
//! - ModelCore for clean model interfaces
//! - WeightLoaderCore for format-agnostic weight loading
//!
//! All models in this module implement the Model trait and use consistent patterns.
// Core model families (completed migration) - WORKING MODELS
pub mod llama;
pub mod qwen;
pub mod gemma;
pub mod phi;
pub mod deepseek;
pub mod mistral;
pub mod mixtral;
// GPT family models
pub mod gpt2;
pub mod gptj;
pub mod gptneox;
pub mod opt;
pub mod bloom;
pub mod mpt;
// Code models
pub mod starcoder;
pub mod codellama;
// Standard decoder models
pub mod olmo;
pub mod granite;
// Additional model families
pub mod yi;
pub mod falcon;
pub mod baichuan;
pub mod internlm;
pub mod chatglm;
pub mod bert;
// Specialized architectures
pub mod t5;
pub mod whisper;
pub mod clip;
pub mod llava;
pub mod mamba;
pub mod minicpm;
// MoE (Mixture of Experts) models
pub mod deepseek_moe;
pub mod dbrx;
pub mod grok;
pub mod arctic;
pub mod jamba;
// RWKV / Linear Attention family
pub mod rwkv4;
pub mod rwkv6;
pub mod recurrent_gemma;
// Vision-Language models
pub mod qwen2_vl;
pub mod phi3_vision;
pub mod internvl;
pub mod cogvlm;
pub mod idefics;
pub mod florence;
// Audio/Speech models
pub mod wav2vec2;
pub mod hubert;
pub mod musicgen;
pub mod encodec;
pub mod traits {
//! Common traits and utilities for V2 model implementations
pub use crate::model_core::*;
pub use crate::tensor_core::*;
pub use crate::weight_loader_core::*;
}