Akshay Ballal 91f1f019b1 Added XLMRobertaModel for Reranking (#2686) ...

* add xlm-roberta-base

* Add task enum for fill-mask and reranker in xlm-roberta example; update README and fix attention mask dimensions

- Introduced a new `Task` enum to replace string task identifiers in the xlm-roberta example.
- Updated the logic in `main.rs` to handle tasks using the new enum.
- Enhanced README with example output for fill-mask task.
- Fixed dimension retrieval in `prepare_4d_attention_mask` function for better clarity and safety.

* Clippy fix.

---------

Co-authored-by: laurent <laurent.mazare@gmail.com>

2024-12-30 11:16:57 +01:00

..

chinese_clip

Add a Context trait similar to anyhow::Context. (#2676)

2024-12-22 09:18:13 +01:00

clip

Add a Context trait similar to anyhow::Context. (#2676)

2024-12-22 09:18:13 +01:00

flux

Module Docs (#2624)

2024-11-18 14:19:23 +01:00

llava

Add a Context trait similar to anyhow::Context. (#2676)

2024-12-22 09:18:13 +01:00

mimi

Module Docs (#2624)

2024-11-18 14:19:23 +01:00

mmdit

Module Docs (#2624)

2024-11-18 14:19:23 +01:00

nvembed_v2

Add Nvembed v2 model (#2649)

2024-12-03 10:56:01 +01:00

openclip

Module Docs (#2624)

2024-11-18 14:19:23 +01:00

pixtral

Fix position encodings for Pixtral (#2678)

2024-12-23 13:22:35 +01:00

segment_anything

More Model Module Docs (#2623)

2024-11-17 20:27:24 +01:00

stable_diffusion

Lint fixes introduced with Rust 1.83 (#2646)

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

whisper

Fix bug in whisper transformer (#2681)

2024-12-24 13:58:21 +01:00

wuerstchen

Module Docs (#2624)

2024-11-18 14:19:23 +01:00

based.rs

Module Docs (#2620)

2024-11-16 09:09:17 +01:00

beit.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

bert.rs

Change/bert encoder public (#2658)

2024-12-04 21:22:30 +01:00

bigcode.rs

Module Docs (#2620)

2024-11-16 09:09:17 +01:00

blip_text.rs

Module Docs (#2624)

2024-11-18 14:19:23 +01:00

blip.rs

Module Docs (#2624)

2024-11-18 14:19:23 +01:00

chatglm.rs

Module Docs (#2624)

2024-11-18 14:19:23 +01:00

codegeex4_9b.rs

Module Docs (#2624)

2024-11-18 14:19:23 +01:00

colpali.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

convmixer.rs

Lint fixes introduced with Rust 1.83 (#2646)

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

convnext.rs

Module Docs (#2624)

2024-11-18 14:19:23 +01:00

dac.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

depth_anything_v2.rs

make DepthAnythingV2 more reusable (#2675)

2024-12-21 12:06:03 +01:00

dinov2.rs

More Model Module Docs (#2623)

2024-11-17 20:27:24 +01:00

dinov2reg4.rs

More Model Module Docs (#2623)

2024-11-17 20:27:24 +01:00

distilbert.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

efficientnet.rs

Add a Context trait similar to anyhow::Context. (#2676)

2024-12-22 09:18:13 +01:00

efficientvit.rs

More Model Module Docs (#2623)

2024-11-17 20:27:24 +01:00

encodec.rs

Lint fixes introduced with Rust 1.83 (#2646)

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

eva2.rs

More Model Module Docs (#2623)

2024-11-17 20:27:24 +01:00

falcon.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

fastvit.rs

Add a Context trait similar to anyhow::Context. (#2676)

2024-12-22 09:18:13 +01:00

gemma2.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

gemma.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

glm4.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

granite.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

hiera.rs

Module Docs (#2624)

2024-11-18 14:19:23 +01:00

jina_bert.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

llama2_c_weights.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

llama2_c.rs

Module Docs (#2624)

2024-11-18 14:19:23 +01:00

llama.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

mamba.rs

Lint fixes introduced with Rust 1.83 (#2646)

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

marian.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

metavoice.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

mistral.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

mixformer.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

mixtral.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

mobileclip.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

mobilenetv4.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

mobileone.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

mod.rs

Added XLMRobertaModel for Reranking (#2686)

2024-12-30 11:16:57 +01:00

moondream.rs

More Model Module Docs (#2623)

2024-11-17 20:27:24 +01:00

mpt.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

olmo.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

paligemma.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

parler_tts.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

persimmon.rs

Module Docs (#2624)

2024-11-18 14:19:23 +01:00

phi3.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

phi.rs

Module Docs (#2624)

2024-11-18 14:19:23 +01:00

quantized_blip_text.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

quantized_blip.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

quantized_llama2_c.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

quantized_llama.rs

Module Docs (#2624)

2024-11-18 14:19:23 +01:00

quantized_metavoice.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

quantized_mistral.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

quantized_mixformer.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

quantized_moondream.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

quantized_mpt.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

quantized_phi3.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

quantized_phi.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

quantized_qwen2.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

quantized_recurrent_gemma.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

quantized_rwkv_v5.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

quantized_rwkv_v6.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

quantized_stable_lm.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

quantized_t5.rs

Module Docs (#2624)

2024-11-18 14:19:23 +01:00

qwen2_moe.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

qwen2.rs

Module Docs (#2624)

2024-11-18 14:19:23 +01:00

recurrent_gemma.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

repvgg.rs

Module Docs (#2624)

2024-11-18 14:19:23 +01:00

resnet.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

rwkv_v5.rs

More Model Module Docs (#2623)

2024-11-17 20:27:24 +01:00

rwkv_v6.rs

More Model Module Docs (#2623)

2024-11-17 20:27:24 +01:00

segformer.rs

Add a Context trait similar to anyhow::Context. (#2676)

2024-12-22 09:18:13 +01:00

siglip.rs

Module Docs (#2624)

2024-11-18 14:19:23 +01:00

stable_lm.rs

Module Docs (#2624)

2024-11-18 14:19:23 +01:00

starcoder2.rs

Module Docs (#2624)

2024-11-18 14:19:23 +01:00

stella_en_v5.rs

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

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

t5.rs

Module Docs (#2624)

2024-11-18 14:19:23 +01:00

trocr.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

vgg.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

vit.rs

Documentation Pass for Models (#2617)

2024-11-15 08:30:15 +01:00

with_tracing.rs

Use the faster rms-norm kernel for llama. (#2107)

2024-04-22 18:52:00 +02:00

xlm_roberta.rs

Added XLMRobertaModel for Reranking (#2686)

2024-12-30 11:16:57 +01:00

yi.rs

Module Docs (#2624)

2024-11-18 14:19:23 +01:00