runtime/models_v2/
mod.rs

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

runtime/models_v2/
mod.rs
