Zack Angelo a2e9d41b20 use softmax_last_dim (metal and cuda kernel) in llama attention layer (#2572)

2024-10-23 20:07:09 +02:00

..

chinese_clip

feat: intergrate chinese clip and add example (#2555)

2024-10-10 15:18:55 +02:00

clip

Add the SigLIP model. (#2515)

2024-09-28 23:48:00 +02:00

flux

Quantized version of flux. (#2500)

2024-09-26 10:23:43 +02:00

llava

Add Pixtral. (#2521)

2024-09-30 19:31:14 +02:00

mimi

Add a RotatingKVCache. (#2493)

2024-09-23 13:14:32 +02:00

mmdit

Add Stable Diffusion 3 Example (#2558)

2024-10-13 22:08:40 +02:00

openclip

MobileCLIP models S1 and S2 (#2454)

2024-08-29 15:38:58 +02:00

pixtral

Pixtral polishing. (#2522)

2024-09-30 21:23:54 +02:00

segment_anything

…

stable_diffusion

Add Stable Diffusion 3 Example (#2558)

2024-10-13 22:08:40 +02:00

whisper

Clippy fixes for 1.81.0. (#2461)

2024-09-05 23:46:55 +02:00

wuerstchen

…

based.rs

Add Based LLM from Hazy Research. (#2411)

2024-08-12 21:21:19 +02:00

beit.rs

…

bert.rs

Add BertForMaskedLM to support SPLADE Models (#2550)

2024-10-07 23:28:21 +02:00

bigcode.rs

Clippy fixes for 1.81.0. (#2461)

2024-09-05 23:46:55 +02:00

blip_text.rs

…

blip.rs

…

chatglm.rs

…

codegeex4_9b.rs

…

colpali.rs

Add ColPali (#2524)

2024-10-01 11:48:39 +02:00

convmixer.rs

…

convnext.rs

…

dac.rs

Add the DAC model. (#2433)

2024-08-19 08:59:51 +02:00

depth_anything_v2.rs

…

dinov2.rs

…

dinov2reg4.rs

…

distilbert.rs

Clippy fixes for 1.81.0. (#2461)

2024-09-05 23:46:55 +02:00

efficientnet.rs

…

efficientvit.rs

…

encodec.rs

Add the DAC model. (#2433)

2024-08-19 08:59:51 +02:00

eva2.rs

…

falcon.rs

Clippy fixes for 1.81.0. (#2461)

2024-09-05 23:46:55 +02:00

fastvit.rs

Add the SigLIP model. (#2515)

2024-09-28 23:48:00 +02:00

gemma2.rs

Add support for gemma-2. (#2425)

2024-08-17 20:31:23 +02:00

gemma.rs

Add ColPali (#2524)

2024-10-01 11:48:39 +02:00

glm4.rs

add models support and example for THUDM/glm-4 (#2362)

2024-08-05 17:48:09 +02:00

granite.rs

Adding Granite 7b Instruct model example (#2487)

2024-09-21 11:52:01 +02:00

hiera.rs

…

jina_bert.rs

Clippy fixes for 1.81.0. (#2461)

2024-09-05 23:46:55 +02:00

llama2_c_weights.rs

…

llama2_c.rs

Clippy fixes for 1.81.0. (#2461)

2024-09-05 23:46:55 +02:00

llama.rs

use softmax_last_dim (metal and cuda kernel) in llama attention layer (#2572)

2024-10-23 20:07:09 +02:00

mamba.rs

…

marian.rs

Enable stable-diffusion 3 on metal. (#2560)

2024-10-14 08:59:12 +02:00

metavoice.rs

…

mistral.rs

Add Pixtral. (#2521)

2024-09-30 19:31:14 +02:00

mixformer.rs

…

mixtral.rs

…

mobileclip.rs

Remove some extra whitelines. (#2513)

2024-09-28 14:41:28 +02:00

mobilenetv4.rs

…

mobileone.rs

…

mod.rs

Adds support for Stella_en_v5 embedding model - 1.5B variant (#2551)

2024-10-13 23:09:12 +02:00

moondream.rs

Clippy fixes for 1.81.0. (#2461)

2024-09-05 23:46:55 +02:00

mpt.rs

…

olmo.rs

…

paligemma.rs

Add ColPali (#2524)

2024-10-01 11:48:39 +02:00

parler_tts.rs

Fix for parler-tts, do not add the last slice of padding tokens. (#2442)

2024-08-22 23:22:03 +02:00

persimmon.rs

…

phi3.rs

…

phi.rs

…

quantized_blip_text.rs

…

quantized_blip.rs

…

quantized_llama2_c.rs

…

quantized_llama.rs

…

quantized_metavoice.rs

…

quantized_mistral.rs

…

quantized_mixformer.rs

…

quantized_moondream.rs

…

quantized_mpt.rs

…

quantized_phi3.rs

…

quantized_phi.rs

…

quantized_qwen2.rs

…

quantized_recurrent_gemma.rs

…

quantized_rwkv_v5.rs

…

quantized_rwkv_v6.rs

…

quantized_stable_lm.rs

…

quantized_t5.rs

…

qwen2_moe.rs

…

qwen2.rs

fix: qwen2 lm_head loading #2443 (#2445)

2024-08-23 16:50:02 +02:00

recurrent_gemma.rs

…

repvgg.rs

…

resnet.rs

…

rwkv_v5.rs

…

rwkv_v6.rs

…

segformer.rs

Clippy fixes for 1.81.0. (#2461)

2024-09-05 23:46:55 +02:00

siglip.rs

Paligemma siglip vision config (#2518)

2024-09-29 17:53:52 +02:00

stable_lm.rs

…

starcoder2.rs

…

stella_en_v5.rs

Adds support for Stella_en_v5 embedding model - 1.5B variant (#2551)

2024-10-13 23:09:12 +02:00

t5.rs

Clippy fixes for 1.81.0. (#2461)

2024-09-05 23:46:55 +02:00

trocr.rs

…

vgg.rs

…

vit.rs

…

with_tracing.rs

…

yi.rs

…