diff options
| author | Akshay Ballal <61191840+akshayballal95@users.noreply.github.com> | 2024-12-30 11:16:57 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-12-30 11:16:57 +0100 |
| commit | 91f1f019b13386f4df3e9b2826c982d10bcc497e (patch) | |
| tree | 1acf1a2921592bbd62ac43ea1321376164b8724b /candle-examples/examples/glm4/main.rs | |
| parent | cd639131f04990c16bfc498ea347cb9df3d2374f (diff) | |
| download | candle-91f1f019b13386f4df3e9b2826c982d10bcc497e.tar.gz candle-91f1f019b13386f4df3e9b2826c982d10bcc497e.tar.bz2 candle-91f1f019b13386f4df3e9b2826c982d10bcc497e.zip | |
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>
Diffstat (limited to 'candle-examples/examples/glm4/main.rs')
0 files changed, 0 insertions, 0 deletions