candle_transformers/models/
mod.rs

1//! Candle implementations for various deep learning models
2//!
3//! This crate provides implementations of popular machine learning models and architectures for different modalities.
4//!
5//!  - Large language models: [`llama`], [`phi3`], [`mamba`], [`mixtral`], [`bert`], ...
6//!  - Text to text models: [`t5`], ...
7//!  - Image to text models: [`blip`], ...
8//!  - Text to image models: [`stable_diffusion`] and [`wuerstchen`], ...
9//!  - Audio models: [`whisper`], [`encodec`], [`metavoice`], [`parler_tts`], ...
10//!  - Computer vision models: [`dinov2`], [`convmixer`], [`efficientnet`], ...
11//!  
12//! Some of the models also have quantized variants, e.g.  [`quantized_blip`], [`quantized_llama`] and  [`quantized_qwen2`].
13//!
14//! The implementations aim to be readable while maintaining good performance. For more information
15//! on each model see the model's module docs in the links below.
16
17pub mod based;
18pub mod beit;
19pub mod bert;
20pub mod bigcode;
21pub mod blip;
22pub mod blip_text;
23pub mod chatglm;
24pub mod chinese_clip;
25pub mod clip;
26pub mod codegeex4_9b;
27pub mod colpali;
28pub mod convmixer;
29pub mod convnext;
30pub mod csm;
31pub mod dac;
32pub mod debertav2;
33pub mod deepseek2;
34pub mod depth_anything_v2;
35pub mod dinov2;
36pub mod dinov2reg4;
37pub mod distilbert;
38pub mod efficientnet;
39pub mod efficientvit;
40pub mod encodec;
41pub mod eva2;
42pub mod falcon;
43pub mod fastvit;
44pub mod flux;
45pub mod gemma;
46pub mod gemma2;
47pub mod gemma3;
48pub mod glm4;
49pub mod glm4_new;
50pub mod granite;
51pub mod granitemoehybrid;
52pub mod helium;
53pub mod hiera;
54pub mod jina_bert;
55pub mod llama;
56pub mod llama2_c;
57pub mod llama2_c_weights;
58pub mod llava;
59pub mod mamba;
60pub mod mamba2;
61pub mod marian;
62pub mod metavoice;
63pub mod mimi;
64pub mod mistral;
65pub mod mixformer;
66pub mod mixtral;
67pub mod mmdit;
68pub mod mobileclip;
69pub mod mobilenetv4;
70pub mod mobileone;
71pub mod modernbert;
72pub mod moondream;
73pub mod mpt;
74pub mod nomic_bert;
75pub mod nvembed_v2;
76pub mod olmo;
77pub mod olmo2;
78pub mod openclip;
79pub mod paddleocr_vl;
80pub mod paligemma;
81pub mod parler_tts;
82pub mod persimmon;
83pub mod phi;
84pub mod phi3;
85pub mod pixtral;
86pub mod quantized_blip;
87pub mod quantized_blip_text;
88pub mod quantized_gemma3;
89pub mod quantized_glm4;
90pub mod quantized_lfm2;
91pub mod quantized_llama;
92pub mod quantized_llama2_c;
93pub mod quantized_metavoice;
94pub mod quantized_mistral;
95pub mod quantized_mixformer;
96pub mod quantized_moondream;
97pub mod quantized_mpt;
98pub mod quantized_phi;
99pub mod quantized_phi3;
100pub mod quantized_qwen2;
101pub mod quantized_qwen3;
102pub mod quantized_qwen3_moe;
103pub mod quantized_recurrent_gemma;
104pub mod quantized_rwkv_v5;
105pub mod quantized_rwkv_v6;
106pub mod quantized_stable_lm;
107pub mod quantized_t5;
108pub mod qwen2;
109pub mod qwen2_moe;
110pub mod qwen3;
111pub mod qwen3_moe;
112pub mod qwen3_vl;
113pub mod recurrent_gemma;
114pub mod repvgg;
115pub mod resnet;
116pub mod rwkv_v5;
117pub mod rwkv_v6;
118pub mod rwkv_v7;
119pub mod segformer;
120pub mod segment_anything;
121pub mod siglip;
122pub mod smol;
123pub mod snac;
124pub mod stable_diffusion;
125pub mod stable_lm;
126pub mod starcoder2;
127pub mod stella_en_v5;
128pub mod t5;
129pub mod trocr;
130pub mod vgg;
131pub mod vit;
132pub mod voxtral;
133pub mod whisper;
134pub mod with_tracing;
135pub mod wuerstchen;
136pub mod xlm_roberta;
137pub mod yi;
138pub mod z_image;
candle_transformers/models/mod.rs

candle_transformers/models/
mod.rs
