Instructions to use codefuse-ai/ML-Embed-0.6B with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use codefuse-ai/ML-Embed-0.6B with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="codefuse-ai/ML-Embed-0.6B")# Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("codefuse-ai/ML-Embed-0.6B") model = AutoModel.from_pretrained("codefuse-ai/ML-Embed-0.6B", device_map="auto") - sentence-transformers
How to use codefuse-ai/ML-Embed-0.6B with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("codefuse-ai/ML-Embed-0.6B") sentences = [ "The weather is lovely today.", "It's so sunny outside!", "He drove to the stadium." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [3, 3] - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -100,7 +100,7 @@ datasets:
|
|
| 100 |
|
| 101 |
# ML-Embed-0.6B
|
| 102 |
|
| 103 |
-
**ML-Embed-0.6B** is a multilingual text embedding model developed by CodeFuse AI and trained from
|
| 104 |
|
| 105 |
This model is designed to be:
|
| 106 |
|
|
@@ -379,21 +379,6 @@ The model is fully open:
|
|
| 379 |
- Training data is available at: [`codefuse-ai/F2LLM-v2`](https://huggingface.co/datasets/codefuse-ai/F2LLM-v2)
|
| 380 |
- Training code is available at: https://github.com/codefuse-ai/CodeFuse-Embeddings/tree/main/ML-Embed
|
| 381 |
|
| 382 |
-
### Citation (to be updated after the conference)
|
| 383 |
-
|
| 384 |
-
```bibtex
|
| 385 |
-
@misc{zhang2026mlembedinclusiveefficientembeddings,
|
| 386 |
-
title={ML-Embed: Inclusive and Efficient Embeddings for a Multilingual World},
|
| 387 |
-
author={Ziyin Zhang and Zihan Liao and Hang Yu and Peng Di and Rui Wang},
|
| 388 |
-
year={2026},
|
| 389 |
-
eprint={2605.15081},
|
| 390 |
-
archivePrefix={arXiv},
|
| 391 |
-
primaryClass={cs.CL},
|
| 392 |
-
url={https://arxiv.org/abs/2605.15081},
|
| 393 |
-
}
|
| 394 |
-
```
|
| 395 |
-
|
| 396 |
-
|
| 397 |
## Intended Uses
|
| 398 |
|
| 399 |
ML-Embed-0.6B can be used for:
|
|
@@ -415,6 +400,20 @@ ML-Embed-0.6B can be used for:
|
|
| 415 |
- For best embedding quality, use appropriate prompts for queries.
|
| 416 |
- Efficient modes such as fewer-layer deployment and low-rank factorized embeddings involve trade-offs between quality and efficiency.
|
| 417 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 418 |
## Related models
|
| 419 |
|
| 420 |
The baseline models are released and submitted to the MTEB leaderboard under the name [`F2LLM-v2`](https://huggingface.co/collections/codefuse-ai/f2llm):
|
|
|
|
| 100 |
|
| 101 |
# ML-Embed-0.6B
|
| 102 |
|
| 103 |
+
**ML-Embed-0.6B** is a multilingual text embedding model developed by CodeFuse AI and trained from Qwen3-0.6B. It is part of the ML-Embed family introduced in the ICML 2026 paper [**ML-Embed: Inclusive and Efficient Embeddings for a Multilingual World**](https://arxiv.org/abs/2605.15081).
|
| 104 |
|
| 105 |
This model is designed to be:
|
| 106 |
|
|
|
|
| 379 |
- Training data is available at: [`codefuse-ai/F2LLM-v2`](https://huggingface.co/datasets/codefuse-ai/F2LLM-v2)
|
| 380 |
- Training code is available at: https://github.com/codefuse-ai/CodeFuse-Embeddings/tree/main/ML-Embed
|
| 381 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 382 |
## Intended Uses
|
| 383 |
|
| 384 |
ML-Embed-0.6B can be used for:
|
|
|
|
| 400 |
- For best embedding quality, use appropriate prompts for queries.
|
| 401 |
- Efficient modes such as fewer-layer deployment and low-rank factorized embeddings involve trade-offs between quality and efficiency.
|
| 402 |
|
| 403 |
+
### Citation (to be updated after the conference)
|
| 404 |
+
|
| 405 |
+
```bibtex
|
| 406 |
+
@misc{zhang2026mlembedinclusiveefficientembeddings,
|
| 407 |
+
title={ML-Embed: Inclusive and Efficient Embeddings for a Multilingual World},
|
| 408 |
+
author={Ziyin Zhang and Zihan Liao and Hang Yu and Peng Di and Rui Wang},
|
| 409 |
+
year={2026},
|
| 410 |
+
eprint={2605.15081},
|
| 411 |
+
archivePrefix={arXiv},
|
| 412 |
+
primaryClass={cs.CL},
|
| 413 |
+
url={https://arxiv.org/abs/2605.15081},
|
| 414 |
+
}
|
| 415 |
+
```
|
| 416 |
+
|
| 417 |
## Related models
|
| 418 |
|
| 419 |
The baseline models are released and submitted to the MTEB leaderboard under the name [`F2LLM-v2`](https://huggingface.co/collections/codefuse-ai/f2llm):
|