mirror of
https://github.com/huggingface/candle.git
synced 2025-06-16 02:38:10 +00:00
0224a749f0
* qwen-moe rebase * lint * fixed rebase error * swapped normal MoE model with CausalMoE Model in example, and swapped the tie word embeddings if statement * updated readme
126 lines
2.9 KiB
Rust
126 lines
2.9 KiB
Rust
//! Candle implementations for various deep learning models
|
|
//!
|
|
//! This crate provides implementations of popular machine learning models and architectures for different modalities.
|
|
//!
|
|
//! - Large language models: [`llama`], [`phi3`], [`mamba`], [`mixtral`], [`bert`], ...
|
|
//! - Text to text models: [`t5`], ...
|
|
//! - Image to text models: [`blip`], ...
|
|
//! - Text to image models: [`stable_diffusion`] and [`wuerstchen`], ...
|
|
//! - Audio models: [`whisper`], [`encodec`], [`metavoice`], [`parler_tts`], ...
|
|
//! - Computer vision models: [`dinov2`], [`convmixer`], [`efficientnet`], ...
|
|
//!
|
|
//! Some of the models also have quantized variants, e.g. [`quantized_blip`], [`quantized_llama`] and [`quantized_qwen2`].
|
|
//!
|
|
//! The implementations aim to be readable while maintaining good performance. For more information
|
|
//! on each model see the model's module docs in the links below.
|
|
|
|
pub mod based;
|
|
pub mod beit;
|
|
pub mod bert;
|
|
pub mod bigcode;
|
|
pub mod blip;
|
|
pub mod blip_text;
|
|
pub mod chatglm;
|
|
pub mod chinese_clip;
|
|
pub mod clip;
|
|
pub mod codegeex4_9b;
|
|
pub mod colpali;
|
|
pub mod convmixer;
|
|
pub mod convnext;
|
|
pub mod csm;
|
|
pub mod dac;
|
|
pub mod debertav2;
|
|
pub mod deepseek2;
|
|
pub mod depth_anything_v2;
|
|
pub mod dinov2;
|
|
pub mod dinov2reg4;
|
|
pub mod distilbert;
|
|
pub mod efficientnet;
|
|
pub mod efficientvit;
|
|
pub mod encodec;
|
|
pub mod eva2;
|
|
pub mod falcon;
|
|
pub mod fastvit;
|
|
pub mod flux;
|
|
pub mod gemma;
|
|
pub mod gemma2;
|
|
pub mod gemma3;
|
|
pub mod glm4;
|
|
pub mod granite;
|
|
pub mod helium;
|
|
pub mod hiera;
|
|
pub mod jina_bert;
|
|
pub mod llama;
|
|
pub mod llama2_c;
|
|
pub mod llama2_c_weights;
|
|
pub mod llava;
|
|
pub mod mamba;
|
|
pub mod marian;
|
|
pub mod metavoice;
|
|
pub mod mimi;
|
|
pub mod mistral;
|
|
pub mod mixformer;
|
|
pub mod mixtral;
|
|
pub mod mmdit;
|
|
pub mod mobileclip;
|
|
pub mod mobilenetv4;
|
|
pub mod mobileone;
|
|
pub mod modernbert;
|
|
pub mod moondream;
|
|
pub mod mpt;
|
|
pub mod nvembed_v2;
|
|
pub mod olmo;
|
|
pub mod olmo2;
|
|
pub mod openclip;
|
|
pub mod paligemma;
|
|
pub mod parler_tts;
|
|
pub mod persimmon;
|
|
pub mod phi;
|
|
pub mod phi3;
|
|
pub mod pixtral;
|
|
pub mod quantized_blip;
|
|
pub mod quantized_blip_text;
|
|
pub mod quantized_gemma3;
|
|
pub mod quantized_llama;
|
|
pub mod quantized_llama2_c;
|
|
pub mod quantized_metavoice;
|
|
pub mod quantized_mistral;
|
|
pub mod quantized_mixformer;
|
|
pub mod quantized_moondream;
|
|
pub mod quantized_mpt;
|
|
pub mod quantized_phi;
|
|
pub mod quantized_phi3;
|
|
pub mod quantized_qwen2;
|
|
pub mod quantized_qwen3;
|
|
pub mod quantized_recurrent_gemma;
|
|
pub mod quantized_rwkv_v5;
|
|
pub mod quantized_rwkv_v6;
|
|
pub mod quantized_stable_lm;
|
|
pub mod quantized_t5;
|
|
pub mod qwen2;
|
|
pub mod qwen2_moe;
|
|
pub mod qwen3;
|
|
pub mod qwen3_moe;
|
|
pub mod recurrent_gemma;
|
|
pub mod repvgg;
|
|
pub mod resnet;
|
|
pub mod rwkv_v5;
|
|
pub mod rwkv_v6;
|
|
pub mod segformer;
|
|
pub mod segment_anything;
|
|
pub mod siglip;
|
|
pub mod snac;
|
|
pub mod stable_diffusion;
|
|
pub mod stable_lm;
|
|
pub mod starcoder2;
|
|
pub mod stella_en_v5;
|
|
pub mod t5;
|
|
pub mod trocr;
|
|
pub mod vgg;
|
|
pub mod vit;
|
|
pub mod whisper;
|
|
pub mod with_tracing;
|
|
pub mod wuerstchen;
|
|
pub mod xlm_roberta;
|
|
pub mod yi;
|