utdawn commited on
Commit
0c40661
Β·
verified Β·
1 Parent(s): abab185

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +155 -3
README.md CHANGED
@@ -1,3 +1,155 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ library_name: transformers
4
+ tags:
5
+ - dllm
6
+ - diffusion
7
+ - llm
8
+ - text_generation
9
+ ---
10
+ # LLaDA2.0-flash
11
+
12
+ **LLaDA2.0-flash** is a diffusion language model featuring a 100BA6B Mixture-of-Experts (MoE) architecture. As an enhanced, instruction-tuned iteration of the LLaDA2.0 series, it is optimized for practical applications.
13
+
14
+ <div align="center">
15
+ <img src="https://mdn.alipayobjects.com/huamei_qa8qxu/afts/img/A*uOo8QKQMiBwAAAAAgNAAAAgAemJ7AQ/original" width="800" />
16
+ </div>
17
+
18
+ ---
19
+
20
+ | Benchmark | Qwen3-30B-A3B-Instruct-2507| Ling-flash-2.0 | LLaDA2.0-flash-preview | LLaDA2.0-flash |
21
+ | :---: | :---: | :---: | :---: | :---: |
22
+ | **Average** | 79.47 | 78.03 | - | 79.32 |
23
+ | **Knowledge** | | | | |
24
+ | MMLU | 87.13 | 87.98 | 83.15 | 87.69 |
25
+ | MMLU-Pro | 74.23 | 76.84 | 49.22 | 73.36 |
26
+ | GPQA | 57.34 | 67.12 | 46.59 | 61.98 |
27
+ | arc-c | 95.81 | 95.08 | | 95.93 |
28
+ | CMMLU | 86.36 | 86.59 | 67.53 | 85.13 |
29
+ | C-EVAL | 88.17 | 88.03 | 66.54 | 86.75 |
30
+ | GAOKAO-Bench | 94.53 | 93.24 | - | 93.90 |
31
+ | **Reasoning** | | | | |
32
+ | SQuAD 2.0 | 89.51 | 81.32 | 85.61 | 90.00 |
33
+ | DROP | 87.57 | 88.32 | 79.49 | 87.90 |
34
+ | KOR-Bench | 68.00 | 68.96 | 37.26 | 64.24 |
35
+ | HellaSwag | 86.31 | 81.59 | - | 84.97 |
36
+ | **Coding** | | | | |
37
+ | CRUXEval-O | 86.75 | 82.75 | 61.88 | 85.12 |
38
+ | MBPP | 86.65 | 85.01 | 77.75 | 88.29 |
39
+ | MultiPL-E | 70.67 | 65.76 | 62.43 | 74.87 |
40
+ | HumanEval | 93.29 | 85.98 | 80.49 | 94.51 |
41
+ | Bigcodebench-Full | 41.49 | 40.70 | 30.44 | 41.58 |
42
+ | LiveCodeBench | 41.63 | 44.11 | 28.58 | 42.29 |
43
+ | Spider | 81.79 | 80.58 | - | 82.49 |
44
+ | **Math** | | | | |
45
+ | GSM8K | 96.36 | 95.45 | 89.01 | 96.06 |
46
+ | MATH | 96.70 | 96.1 | 73.50 | 95.44 |
47
+ | OlympiadBench | 77.59 | 76.19 | 47.78 | 74.07 |
48
+ | AIME 2025 | 61.88 | 55.89 | - | 60.00 |
49
+ | **Agent & Alignment** | | | | |
50
+ | BFCL_Live | 73.19 | 67.57 | 74.11 | 75.43 |
51
+ | IFEval-strict -prompt | 84.29 | 81.52 | 62.50 | 81.70 |
52
+
53
+
54
+
55
+ ## πŸš€ Performance Highlights
56
+ + **Leading MoE Architecture**:
57
+ The open-source **Mixture-of-Experts (MoE) diffusion large language model**, pre-trained from scratch on approximately **20 trillion tokens**.
58
+ + **Efficient Inference**:
59
+ With **100 billion total parameters**, only **6.1 billion** are activated during inference. LLaDA2.0-flash significantly reduces computational costs while outperforming open-source dense models of similar scale.
60
+ + **Impressive Performance on Code & Complex Reasoning**:
61
+ Excels in tasks such as **code generation** and **advanced mathematical reasoning**, demonstrating strong reasoning capabilities.
62
+ + **Tool Use**:
63
+ Supports **tool calling** and achieves excellent performance in complex agent-based tasks.
64
+ + **Open & Extensible**:
65
+ Fully open-source with commitment to transparency. We plan to release a **leading inference framework** in the future and continue investing in cutting-edge areas like **diffusion LLMs (dLLM)** to drive disruptive innovation.
66
+
67
+ ## πŸ—ΊοΈ What's Next
68
+
69
+ + **Supercharged Reasoning with LLaDA 2.0:** LLaDA 2.0 series will be fine-tuned with **Reinforcement Learning**, unlocking a new level of sophisticated reasoning and problem-solving abilities.
70
+ + **Tools for Innovators:** The model was finetuned on the [dFactory](https://github.com/inclusionAI/dFactory) framework using Fully Sharded Data Parallel (FSDP2). We have begun open-sourcing dFactory and will continuously release our advanced post-training technologies. Whether you want to master the current model or build your own customized versions, you'll have the tools you need. Stay tuned for more updates!
71
+
72
+ ---
73
+
74
+ ## πŸ“¦ Model Variants
75
+ | Model ID | Description | Hugging Face Link |
76
+ | --- | --- | --- |
77
+ | `inclusionAI/LLaDA2.0-mini` | Instruction-tuned model, ready for downstream applications. | [πŸ€— Model Card](https://huggingface.co/inclusionAI/LLaDA2.0-mini) |
78
+ | `inclusionAI/LLaDA2.0-flash` | Instruction-tuned model, ready for downstream applications. | [πŸ€— Model Card](https://huggingface.co/inclusionAI/LLaDA2.0-flash) |
79
+
80
+
81
+ ---
82
+
83
+ ## πŸ” Model Overview
84
+ **LLaDA2.0-flash** has the following specifications:
85
+
86
+ + **Type**: Mixture-of-Experts (MoE) Diffusion Language Model
87
+ + **Total Parameters (Non-Embedding)**: 100B
88
+ + **Number of Layers**: 32
89
+ + **Attention Heads**: 32
90
+ + **Context Length**: 32,768 tokens
91
+ + **Position Embedding**: Rotary (RoPE)
92
+ + **Vocabulary Size**: 157,184
93
+
94
+ ---
95
+
96
+ ### πŸ€— Hugging Face Transformers
97
+ Make sure you have `transformers` and its dependencies installed:
98
+
99
+ ```python
100
+ import torch
101
+ import torch.nn.functional as F
102
+ from transformers import AutoModelForCausalLM
103
+ from transformers import AutoTokenizer
104
+
105
+ model_path = "/path/to/LLaDA2.0-mini-preview"
106
+ device = "auto"
107
+ model = AutoModelForCausalLM.from_pretrained(
108
+ model_path, trust_remote_code=True, device_map=device
109
+ )
110
+ model = model.to(torch.bfloat16)
111
+ model.eval()
112
+ tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
113
+
114
+ prompt = "Why does Camus think that Sisyphus is happy?"
115
+ input_ids = tokenizer.apply_chat_template(
116
+ [{"role": "user", "content": prompt}],
117
+ add_generation_prompt=True,
118
+ tokenize=True,
119
+ return_tensors="pt",
120
+ )
121
+ generated_tokens = model.generate(
122
+ inputs=input_ids,
123
+ eos_early_stop=True,
124
+ gen_length=512,
125
+ block_length=32,
126
+ steps=32,
127
+ temperature=0.0,
128
+ )
129
+ generated_answer = tokenizer.decode(
130
+ generated_tokens[0],
131
+ skip_special_tokens=True,
132
+ )
133
+ print(generated_answer)
134
+ ```
135
+
136
+ ### Best Practices
137
+ To achieve optimal performance, we recommend the following settings:
138
+
139
+ 1. **Sampling Parameters**:
140
+ We suggest using `Temperature=0.0`, `block_length=32`, and `steps=32`. Using a higher temperature value may occasionally result in language mixing and a slight decrease in model performance.
141
+
142
+ 2. **Adequate Output Length**:
143
+ We recommend using an output length of 32768 tokens for most queries.
144
+
145
+ ---
146
+
147
+ ## 🌐 License
148
+ This project is licensed under the terms of the [Apache License 2.0](https://www.apache.org/licenses/LICENSE-2.0).
149
+
150
+ ---
151
+
152
+ ## 🀝 Contact & Collaboration
153
+ For questions, collaborations, or feedback, please reach out via [Hugging Face](https://huggingface.co/inclusionAI/LLaDA2.0-flash) or open an issue in the [repository](https://github.com/inclusionAI).
154
+
155
+ πŸ‘‰ Join us in advancing open, efficient, and intelligent language models!