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 dac;
31pub mod debertav2;
32pub mod deepseek2;
33pub mod depth_anything_v2;
34pub mod dinov2;
35pub mod dinov2reg4;
36pub mod distilbert;
37pub mod efficientnet;
38pub mod efficientvit;
39pub mod encodec;
40pub mod eva2;
41pub mod falcon;
42pub mod fastvit;
43pub mod flux;
44pub mod gemma;
45pub mod gemma2;
46pub mod gemma3;
47pub mod glm4;
48pub mod granite;
49pub mod helium;
50pub mod hiera;
51pub mod jina_bert;
52pub mod llama;
53pub mod llama2_c;
54pub mod llama2_c_weights;
55pub mod llava;
56pub mod mamba;
57pub mod marian;
58pub mod metavoice;
59pub mod mimi;
60pub mod mistral;
61pub mod mixformer;
62pub mod mixtral;
63pub mod mmdit;
64pub mod mobileclip;
65pub mod mobilenetv4;
66pub mod mobileone;
67pub mod modernbert;
68pub mod moondream;
69pub mod mpt;
70pub mod nvembed_v2;
71pub mod olmo;
72pub mod openclip;
73pub mod paligemma;
74pub mod parler_tts;
75pub mod persimmon;
76pub mod phi;
77pub mod phi3;
78pub mod pixtral;
79pub mod quantized_blip;
80pub mod quantized_blip_text;
81pub mod quantized_llama;
82pub mod quantized_llama2_c;
83pub mod quantized_metavoice;
84pub mod quantized_mistral;
85pub mod quantized_mixformer;
86pub mod quantized_moondream;
87pub mod quantized_mpt;
88pub mod quantized_phi;
89pub mod quantized_phi3;
90pub mod quantized_qwen2;
91pub mod quantized_recurrent_gemma;
92pub mod quantized_rwkv_v5;
93pub mod quantized_rwkv_v6;
94pub mod quantized_stable_lm;
95pub mod quantized_t5;
96pub mod qwen2;
97pub mod qwen2_moe;
98pub mod recurrent_gemma;
99pub mod repvgg;
100pub mod resnet;
101pub mod rwkv_v5;
102pub mod rwkv_v6;
103pub mod segformer;
104pub mod segment_anything;
105pub mod siglip;
106pub mod stable_diffusion;
107pub mod stable_lm;
108pub mod starcoder2;
109pub mod stella_en_v5;
110pub mod t5;
111pub mod trocr;
112pub mod vgg;
113pub mod vit;
114pub mod whisper;
115pub mod with_tracing;
116pub mod wuerstchen;
117pub mod xlm_roberta;
118pub mod yi;
candle_transformers/models/mod.rs

candle_transformers/models/
mod.rs
