LLaDA2.1-mini
π LLaDA2.1-flash is now live on ZenmuxAI! Try it via API π οΈ or Chat π¬: https://zenmux.ai/inclusionai/llada2.1-flash
LLaDA2.1-mini is a diffusion language model of the LLaDA series featuring the editing enhancement. It significantly improves inference speed while delivering strong task performance.
Model Performance
π Highlights
- Error-Correcting Editable: Structural innovation of editable generation for dLLM
- Speedy vs Quality Mode: The 16B mini model achieves ultra-fast inference under Speed Mode while remaining competitive across various tasks and under Quality Mode.
- Reinforcement Learning on 100B-scale dLLM: Tailored algorithm and framework to enable reinforcement learning for large dLLM.
πΊοΈ What's Next
- Powerful Agentic/Tool Use Capability with LLaDA: Next update will be equipped with powerful Agentic and long-distance tool-use capability.
- Extreme Editing: Next update will feature stronger and more extensive editing capabilities, aimed at correcting more errors in parallel reasoning.
- Explore More Training Paradigms: We want to explore more training paradigms than SFT and RL for dLLM.
π¦ Model Variants
π Model Overview
LLaDA2.1-mini has the following specifications:
- Type: Mixture-of-Experts (MoE) Diffusion Language Model
- Total Parameters (Non-Embedding): 16B
- Number of Layers: 20
- Attention Heads: 16
- Context Length: 32,768 tokens
- Position Embedding: Rotary (RoPE)
- Vocabulary Size: 157,184
π€ Hugging Face Transformers
Make sure you have transformers and its dependencies installed:
Best Practices
To achieve optimal performance, we recommend the following settings:
-
Sampling Parameters:
We recommend the following general sampling parameters: block_length=32, temperature=0.0, top_p=None and top_k=None. We are currently exploring more diverse sampling configurations.
-
Denoising Thresholds:
There are three denoising params: threshold, editing_threshold and max_post_steps. We recommend threshold=0.7, editing_threshold=0.5 for Quality Mode and threshold=0.5, editing_threshold=0.0 for Speed Mode. For both modes, we suggest setting max_post_steps to a value greater than 5. We recommend 16 as a balanced default, which was used for most of our internal testing.
Note: Low threshold may causes stuttering in trade-off for quick inference.
- Adequate Output Length:
We recommend using an output length of 16384 tokens for most scenarios.
π€ModelScope
If you're in mainland China, we strongly recommend you to use our model from π€ModelScope
Deployment
SGLang
SGLang enables dLLM inference either through offline batching or by launching an HTTP server for online requests. You can start the SGLang dLLM using the following commands:
Enviroment Preparation
Pull Request (PR) has been submitted and merged to the SGLang community, please prepare the environment with the lateset version
π License
This project is licensed under the terms of the Apache License 2.0.
π€ Contact & Collaboration
For questions, collaborations, or feedback, please reach out via Hugging Face or open an issue in the repository.
π Join us in advancing open, efficient, and intelligent language models!
Citation