Hugging Face's logo

KaedeTai
/
dflash2-mlx-quantized-draft

MLX
qwen3
speculative-decoding
dflash2
apple-silicon
quantization
benchmark
4-bit precision
Model card Files
xet
 Community
dflash2-mlx-quantized-draft
File size: 1,274 Bytes
78c877d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
{
 "architectures": [
  "DFlash2DraftModel"
 ],
 "attention_bias": false,
 "attention_dropout": 0.0,
 "bos_token_id": null,
 "is_causal": false,
 "dflash_config": {
  "block_size": 8,
  "conv_group_size": 16,
  "conv_kernel_size": 2,
  "mask_token_id": 248070,
  "selector_rank": 256,
  "selector_top_k": 16,
  "target_layer_ids": [
   5,
   19,
   33,
   47,
   61
  ]
 },
 "dtype": "bfloat16",
 "eos_token_id": 248044,
 "head_dim": 128,
 "hidden_act": "silu",
 "hidden_size": 5120,
 "initializer_range": 0.02,
 "intermediate_size": 17408,
 "layer_types": [
  "sliding_attention",
  "sliding_attention",
  "sliding_attention",
  "sliding_attention",
  "sliding_attention"
 ],
 "max_position_embeddings": 262144,
 "max_window_layers": 5,
 "model_type": "qwen3",
 "num_attention_heads": 32,
 "num_hidden_layers": 5,
 "num_key_value_heads": 8,
 "num_target_layers": 64,
 "pad_token_id": 248044,
 "rms_norm_eps": 1e-06,
 "rope_parameters": {
  "rope_theta": 10000000,
  "rope_type": "default"
 },
 "sliding_window": 2048,
 "tie_word_embeddings": false,
 "transformers_version": "5.15.0",
 "use_cache": true,
 "use_sliding_window": true,
 "vocab_size": 248320,
 "quantization": {
  "group_size": 64,
  "bits": 4
 },
 "quantization_config": {
  "group_size": 64,
  "bits": 4
 }
}