huggingface / huggingface/candle

Model Wishlist

Open
#1,177 107 comments 11 reactions 0 assignees View on GitHub
help wanted
Dominant language
Rust
Stars
21k
Forks
1.8k
Avg merge
16h 42m
Merged PRs (30d)
25

Description

This issue aims at keeping track of the models that would be interesting to get added to candle. Feel free to make a comment to mention a new model, or vote for a model already in the list.
- [musicgen](https://huggingface.co/facebook/musicgen-small).
- [nougat](https://huggingface.co/facebook/nougat-base), [donut](https://huggingface.co/docs/transformers/model_doc/donut), see #1070.
- [LCM Dreamshaper v7](https://huggingface.co/SimianLuo/LCM_Dreamshaper_v7), fast diffusion model.

Added recently:
- [JinaBert](https://huggingface.co/jinaai/jina-embeddings-v2-base-en), embeddings model with an apache 2 license and a larger context (+ using alibi biases rather than rope embeddings).
- [Marian-MT](https://github.com/huggingface/candle/tree/main/candle-examples/examples/marian-mt), a neural machine translation model.
- [Yi-6b / Yi-34b](https://github.com/huggingface/candle/blob/main/candle-examples/examples/yi/main.rs), bilingual (English/Chinese) LLM.

Contributor guide

No contributing guide indexed for this repository

Research direction

This is a broad model wishlist rather than a scoped implementation task. Start by reviewing the linked candle-examples/examples/marian-mt and candle-examples/examples/yi/main.rs entries, then read the comments to identify a model with clear consensus. Done would require an agreed model and implementation scope beyond this issue.

Written by the indexing model from the issue text.

Assessment

Tech stack
huggingface, rust
Domain
machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.