Laurent Mazare a0460cd2b1 Add the code-gemma models. (#2038) ...

* Add the code-gemma models.

* Tweak to the gemma config.

2024-04-10 21:19:21 +02:00

..

clip

Fix clippy lints + minor cleanups. (#1957)

2024-03-28 14:17:46 +01:00

segment_anything

Relax the contiguous check for cuda kernels. (#2000)

2024-04-03 09:02:38 +02:00

stable_diffusion

Improve the handling of matmul with squeezed layouts. (#1998)

2024-04-02 23:17:05 +02:00

whisper

Speaker embeddings computation for metavoice. (#1800)

2024-03-04 14:13:01 +01:00

wuerstchen

…

bert.rs

…

bigcode.rs

…

blip_text.rs

…

blip.rs

…

chatglm.rs

…

convmixer.rs

…

convnext.rs

…

dinov2.rs

Another fix for squeezing. (#1943)

2024-03-26 17:05:26 +01:00

distilbert.rs

…

efficientnet.rs

…

efficientvit.rs

EfficientVit (MSRA) model (#1783)

2024-03-01 08:53:52 +01:00

encodec.rs

Add a flag to select the dtype used in metavoice. (#1805)

2024-03-05 12:16:00 +01:00

falcon.rs

Support alternative dtypes for mamba (#2036)

2024-04-10 18:10:01 +02:00

gemma.rs

Add the code-gemma models. (#2038)

2024-04-10 21:19:21 +02:00

jina_bert.rs

…

llama2_c_weights.rs

…

llama2_c.rs

Avoid the attention mask where possible. (#1933)

2024-03-25 15:31:04 +01:00

llama.rs

Also avoid the mask in the llama example.

2024-03-24 19:04:32 +01:00

mamba.rs

Support alternative dtypes for mamba (#2036)

2024-04-10 18:10:01 +02:00

marian.rs

…

metavoice.rs

Add some tracing to metavoice. (#1826)

2024-03-09 12:24:11 +01:00

mistral.rs

Use the new rope kernel in mistral. (#1937)

2024-03-25 23:26:05 +01:00

mixformer.rs

Faster mask implementation for mixformers. (#2017)

2024-04-05 09:38:26 +02:00

mixtral.rs

Use a common with_tracing::RmsNorm in a few models. (#1871)

2024-03-18 21:40:06 +01:00

mobileone.rs

…

mod.rs

Quantized moondream implementation and BOS token (#1980)

2024-04-01 19:37:54 +02:00

moondream.rs

Moondream tracing. (#2016)

2024-04-05 09:11:08 +02:00

mpt.rs

…

persimmon.rs

…

phi.rs

…

quantized_blip_text.rs

…

quantized_blip.rs

…

quantized_llama2_c.rs

Avoid the attention mask where possible. (#1933)

2024-03-25 15:31:04 +01:00

quantized_llama.rs

Remove some unnecessary calls to contiguous. (#1968)

2024-03-30 13:22:00 +01:00

quantized_metavoice.rs

Fix the final rmsnorm for quantized-metavoice. (#2021)

2024-04-06 19:35:01 +02:00

quantized_mistral.rs

Use the new rope kernel in mistral. (#1937)

2024-03-25 23:26:05 +01:00

quantized_mixformer.rs

Quantized moondream implementation and BOS token (#1980)

2024-04-01 19:37:54 +02:00

quantized_moondream.rs

Quantized moondream implementation and BOS token (#1980)

2024-04-01 19:37:54 +02:00

quantized_mpt.rs

…

quantized_rwkv_v5.rs

add quantized rwkv v5 model (#1743)

2024-02-25 21:43:40 +01:00

quantized_rwkv_v6.rs

add models of rwkv v6 and quantized rwkv v6 (#1781)

2024-03-01 08:37:56 +01:00

quantized_stable_lm.rs

Rustfmt fix. (#1788)

2024-03-02 10:35:07 +01:00

quantized_t5.rs

…

qwen2_moe.rs

Qwen MoE model. (#1960)

2024-03-28 23:10:57 +01:00

qwen2.rs

Use a common with_tracing::RmsNorm in a few models. (#1871)

2024-03-18 21:40:06 +01:00

repvgg.rs

…

resnet.rs

…

rwkv_v5.rs

add quantized rwkv v5 model (#1743)

2024-02-25 21:43:40 +01:00

rwkv_v6.rs

add models of rwkv v6 and quantized rwkv v6 (#1781)

2024-03-01 08:37:56 +01:00

segformer.rs

Add an initial Segformer implementation (#1617)

2024-03-03 16:01:46 +01:00

stable_lm.rs

Rustfmt fix. (#1788)

2024-03-02 10:35:07 +01:00

starcoder2.rs

Add the StarCoder2 model. (#1779)

2024-02-28 21:02:41 +01:00

t5.rs

Expose the t5 config fields + allow t5-large. (#1987)

2024-04-01 20:58:34 +02:00

trocr.rs

…

vgg.rs

…

vit.rs

…

with_tracing.rs

Use a common with_tracing::RmsNorm in a few models. (#1871)

2024-03-18 21:40:06 +01:00

yi.rs

Use a common with_tracing::RmsNorm in a few models. (#1871)

2024-03-18 21:40:06 +01:00