Views
No views yet
GFusion-10B-A1.8B is an experimental instruction-tuned diffusion language model trained by adapting GigaChat3-10B-A1.8B-base to block diffusion generation, followed by context extension, SFT, and confidence tuning.GigaChat3-10B-A1.8B-base model card.
| Benchmark | GFusion 10B-A1.8B | GFusion + CT 10B-A1.8B | GigaChat3 10B-A1.8B | LLaDA-MoE 7B-A1.4B | LLaDA2.0-mini preview 16B-A1.4B |
|---|---|---|---|---|---|
| MMLU | 73.38 | 73.09 | 71.20 | 67.18 | 72.49 |
| MMLU-Pro | 58.48 | 58.04 | 59.60 | 44.64 | 49.22 |
| IFEval | 70.38 | 71.22 | 66.55 | 59.33 | 62.50 |
| GPQA | 33.84 | 32.12 | 35.02 | -- | 23.74 |
| TruthfulQA | 44.84 | 44.68 | 45.90 | -- | 56.54 |
| GSM8K | 84.48 | 83.78 | 85.44 | 82.41 | 89.01 |
| MGSM | 78.80 | 79.20 | 76.80 | -- | 81.44 |
| MATH | 68.08 | 66.86 | 70.00 | 58.68 | 73.50 |
| MBPP+ | 67.20 | 65.81 | 63.60 | -- | 66.67 |
| HumanEval | 75.00 | 71.34 | 72.56 | 61.59 | 80.49 |
| HumanEval+ | 65.63 | 63.63 | 66.46 | -- | 71.95 |
| LCB-Lite | 29.10 | 29.09 | 31.94 | -- | 29.07 |
| RUBQ | 63.49 | 62.56 | 65.16 | -- | 16.84 |
| MMLU-RU | 67.92 | 67.74 | 66.20 | -- | 50.48 |
| IFEval-RU | 61.27 | 64.51 | 64.19 | -- | 55.75 |
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3device = "auto"
4model_path = "ai-sage/GFusion-10B-A1.8B-bf16"
5
6model = AutoModelForCausalLM.from_pretrained(
7 model_path, device_map=device, trust_remote_code=True
8)
9tokenizer = AutoTokenizer.from_pretrained(
10 model_path, device_map=device, trust_remote_code=True
11)
12
13messages = [
14 {"role": "user", "content": "What are the KKT optimality conditions?"}
15]
16inputs = tokenizer.apply_chat_template(
17 messages, add_generation_prompt=True, return_tensors="pt"
18).to(model.device)
19
20outputs = model.generate(
21 inputs,
22 max_new_tokens=512,
23 block_size=32,
24 gamma=0.70
25)
26
27print(tokenizer.decode(outputs[0], skip_special_tokens=True))1git clone https://github.com/sgl-project/sglang.git
2cd sglang
3
4git fetch origin refs/pull/29776/head:gfusion
5git switch gfusion1curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y && source "$HOME/.cargo/env"
2pip install --upgrade pip
3
4######### CUDA 13 #########
5pip install -e "python[all]"
6
7######### CUDA 12 #########
8# 12.6 -> CU=126, KW=cu124
9# 12.8 -> CU=128, KW=cu129
10# 12.9 -> CU=129, KW=cu129
11
12CU=128 ; KW=cu129
13pip install --no-deps --force-reinstall \
14 "https://github.com/sgl-project/whl/releases/download/v0.4.4/sglang_kernel-0.4.4+${KW}-cp310-abi3-manylinux2014_$(uname -m).whl"
15
16# sglang + deps
17pip install --extra-index-url "https://download.pytorch.org/whl/cu${CU}" -e "python[all]"
18
19# re-pin the wheels to the cu12 build
20pip list --format=freeze | awk -F'==' '/-cu13(==|$)/ {print $1}' | xargs -r pip uninstall -y
21pip install --index-url "https://download.pytorch.org/whl/cu${CU}" --force-reinstall \
22 torch==2.11.0 torchvision==0.26.0 torchaudio==2.11.0
23pip install --no-deps --force-reinstall \
24 "https://github.com/sgl-project/whl/releases/download/v0.1.3/sgl_deep_gemm-0.1.3+cu129-py3-none-manylinux2014_$(uname -m).whl"1# eb_sampling.yaml
2gamma: 0.151python -m sglang.launch_server \
2 --model-path ai-sage/GFusion-10B-A1.8B-bf16 \
3 --dllm-algorithm EBSampling \
4 --dllm-algorithm-config eb_sampling.yaml \
5 --attention-backend <fa3 or triton> \
6 --host 0.0.0.0 \
7 --port 30000 \
8 --dtype float16 \
9 --mem-fraction-static 0.88 \
10 --cuda-graph-bs-decode 11curl http://localhost:30000/v1/chat/completions \
2 -H "Content-Type: application/json" \
3 -d '{
4 "model": "ai-sage/GFusion-10B-A1.8B-bf16",
5 "temperature": 0,
6 "max_tokens": 512,
7 "messages": [
8 {
9 "role": "user",
10 "content": "What are the KKT optimality conditions?"
11 }
12 ]
13 }'