Blockway-team commited on
Commit
2fbf6e5
·
verified ·
1 Parent(s): f4f2041

Random-weight tiny checkpoint for CI (no trained weights)

Browse files
Files changed (4) hide show
  1. README.md +11 -0
  2. SHA256SUMS +3 -0
  3. config.json +50 -0
  4. model.safetensors +3 -0
README.md ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags: [volundr, dflash, test, random-weights]
4
+ ---
5
+
6
+ # Agens-Volundr-DFlash2-tiny-random
7
+
8
+ **Random weights, for CI only.** A 1-layer DFlash2 drafter (block size 8, selector rank 16, target
9
+ layers 0 and 2) for `Blockway/Agens-Volundr-tiny-random`, with the same DFlash2 config fields as
10
+ `Blockway/Agens-Volundr-32B-v1-DFlash2`. Accept lengths are meaningless; it exercises the
11
+ speculative-decoding code paths in the SGLang Volundr tests.
SHA256SUMS ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ eaf678083e277bd39977ec8d3cff76d378b6a91fccf5978c79f1fd70f0867008 config.json
2
+ 8e97809d6e7f186ec1c31dd2cb6cb347070dc4e5badf19ca589dc898775c20d7 model.safetensors
3
+ 4ad01364080c7db03022e5ee08578cf4a41d4e91ef2cd1830510a876af2f6aff README.md
config.json ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "DFlash2DraftModel"
4
+ ],
5
+ "model_type": "llama",
6
+ "is_causal": false,
7
+ "attention_bias": false,
8
+ "attention_dropout": 0.0,
9
+ "bos_token_id": null,
10
+ "dflash_config": {
11
+ "block_size": 8,
12
+ "conv_group_size": 16,
13
+ "conv_kernel_size": 2,
14
+ "mask_token_id": 248070,
15
+ "selector_rank": 16,
16
+ "selector_top_k": 4,
17
+ "target_layer_ids": [
18
+ 0,
19
+ 2
20
+ ]
21
+ },
22
+ "dtype": "bfloat16",
23
+ "eos_token_id": 248046,
24
+ "head_dim": 64,
25
+ "hidden_act": "silu",
26
+ "hidden_size": 64,
27
+ "initializer_range": 0.02,
28
+ "intermediate_size": 256,
29
+ "layer_types": [
30
+ "sliding_attention"
31
+ ],
32
+ "max_position_embeddings": 32768,
33
+ "max_window_layers": 1,
34
+ "num_attention_heads": 2,
35
+ "num_hidden_layers": 1,
36
+ "num_key_value_heads": 1,
37
+ "num_target_layers": 4,
38
+ "pad_token_id": 248044,
39
+ "rms_norm_eps": 1e-06,
40
+ "rope_parameters": {
41
+ "rope_theta": 10000000,
42
+ "rope_type": "default"
43
+ },
44
+ "sliding_window": 2048,
45
+ "tie_word_embeddings": false,
46
+ "transformers_version": "5.15.0",
47
+ "use_cache": true,
48
+ "use_sliding_window": true,
49
+ "vocab_size": 248320
50
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8e97809d6e7f186ec1c31dd2cb6cb347070dc4e5badf19ca589dc898775c20d7
3
+ size 16066432