iskng 4f59ed38b0 Adds support for stella_en_v5 embedding model -400M variant (#2608) ...

* Adds support for stella_en_v5 embedding model -400M variant

* Unified stella

* WIP: Unified Stella

* Combined stella for both 1.5B and 400M variants

* Cargo fmt for the CI

* removed redundant stella-400m model and example after merge into stella-en-v5

* cargo fmt --all

---------

Co-authored-by: Anubhab Bandyopadhyay <4890833+AnubhabB@users.noreply.github.com>
Co-authored-by: laurent <laurent.mazare@gmail.com>

2024-11-29 09:01:08 +01:00

..

based

Add Based LLM from Hazy Research. (#2411)

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

beit

…

bert

bert attention mask (#1934)

2024-08-01 08:26:19 +02:00

bigcode

Adapt more examples to the updated safetensor api. (#947)

2023-09-23 21:26:03 +01:00

blip

…

chatglm

…

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

codegeex4-9b

…

colpali

Add ColPali (#2524)

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

convmixer

…

convnext

…

custom-ops

…

depth_anything_v2

…

dinov2

…

dinov2reg4

…

distilbert

…

efficientnet

…

efficientvit

…

encodec

Make the RNN configs accessible from the models. (#2541)

2024-10-04 14:22:23 +02:00

eva2

…

falcon

…

fastvit

FastViT fixes. (#2452)

2024-08-28 11:20:09 +02:00

flux

Add a seed to the flux example. (#2529)

2024-10-02 10:52:02 +02:00

gemma

Add support for gemma-2. (#2425)

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

glm4

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

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

granite

Adding Granite 7b Instruct model example (#2487)

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

gte-qwen

…

hiera

…

jina-bert

…

llama

Add the SmolLM2 models. (#2595)

2024-11-03 17:11:12 +01:00

llama2-c

…

llama_multiprocess

Apply rustfmt. (#2421)

2024-08-16 18:57:14 +02:00

llava

…

mamba

…

mamba-minimal

Lint fixes introduced with Rust 1.83 (#2646)

2024-11-28 23:00:21 +01:00

marian-mt

Fix the marian tokenizer importer. (#2426)

2024-08-17 20:58:40 +02:00

metavoice

…

mimi

Make the RNN configs accessible from the models. (#2541)

2024-10-04 14:22:23 +02:00

mistral

…

mixtral

…

mnist-training

…

mobileclip

Remove some extra whitelines. (#2513)

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

mobilenetv4

MobileCLIP models S1 and S2 (#2454)

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

mobileone

…

moondream

…

musicgen

Clippy fixes for 1.81.0. (#2461)

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

olmo

…

onnx

…

paligemma

Add PaliGemma. (#2519)

2024-09-29 19:56:56 +02:00

parler-tts

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

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

phi

Enable BF16 on metal. (#2380)

2024-08-01 11:05:07 +02:00

pixtral

Pixtral polishing. (#2522)

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

quantized

Add the SmolLM2 models. (#2595)

2024-11-03 17:11:12 +01:00

quantized-phi

…

quantized-qwen2-instruct

…

quantized-t5

…

qwen

…

recurrent-gemma

…

reinforcement-learning

Remove some unused macros. (#2618)

2024-11-15 16:46:55 +01:00

replit-code

…

repvgg

…

resnet

…

rwkv

…

segformer

…

segment-anything

onnx: fix pad, unsqueeze (#2317)

2024-07-23 23:10:57 +02:00

siglip

Add the SigLIP model. (#2515)

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

silero-vad

Clippy fixes for 1.81.0. (#2461)

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

splade

Add BertForMaskedLM to support SPLADE Models (#2550)

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

stable-diffusion

onnx: fix pad, unsqueeze (#2317)

2024-07-23 23:10:57 +02:00

stable-diffusion-3

Support Skip Layer Guidance (SLG) for Stable Diffusion 3.5 Medium (#2590)

2024-11-01 18:10:40 +01:00

stable-lm

Avoid copying the data on squeeze and unsqueeze. (#1884)

2024-03-20 13:04:36 +01:00

starcoder2

Add the StarCoder2 model. (#1779)

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

stella-en-v5

Adds support for stella_en_v5 embedding model -400M variant (#2608)

2024-11-29 09:01:08 +01:00

t5

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

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

trocr

…

vgg

…

vit

…

whisper

Add required feature for whisper example in Readme (#2539)

2024-10-04 14:29:55 +02:00

whisper-microphone

Fix for whisper-microphone example failure if audio isn't chunk aligned (#2645)

2024-11-27 22:35:11 +01:00

wuerstchen

…

yi

…

yolo-v3

Clippy fixes for 1.81.0. (#2461)

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

yolo-v8

Clippy fixes for 1.81.0. (#2461)

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

onnx_basics.rs

…