Upload folder using huggingface_hub
Browse files- README.md +7 -3
- config.json +3 -55
- model.safetensors +2 -2
README.md
CHANGED
|
@@ -13,8 +13,8 @@ Structural comparison with `RecursiveSI/flint`:
|
|
| 13 |
| Loading implementation | source SGLang plugin | same source plugin wheel |
|
| 14 |
| Stored precision | BF16 | BF16 |
|
| 15 |
| Vocabulary / embedding and LM-head ties | 150,144 / untied | same |
|
| 16 |
-
| Layers / dense prefix | 49 / layers 0-1 |
|
| 17 |
-
| Attention cadence / RoPE-free layers | 49-layer source schedule |
|
| 18 |
| Hidden size / Q and KV heads / head dimension | 2,048 / 32 and 4 / 128 | 32 / 1 and 1 / 32 |
|
| 19 |
| Routed experts / active experts | 256 / 6 | 64 / 6 |
|
| 20 |
| MoE / shared-expert widths | 512 / 1,024 | 32 / 64 |
|
|
@@ -27,7 +27,7 @@ validates TP=1 only, so this model carries no TP=4 claim.
|
|
| 27 |
|
| 28 |
| File path | Size |
|
| 29 |
|------|------|
|
| 30 |
-
| model.safetensors |
|
| 31 |
|
| 32 |
|
| 33 |
### Example usage:
|
|
@@ -71,10 +71,14 @@ raw = json.loads(Path(hf_hub_download(source_model_id, "config.json", revision=s
|
|
| 71 |
# The fixed 150,144-row untied embedding and LM head make hidden_size=32 the
|
| 72 |
# smallest practical BF16 geometry; one 32-wide attention head retains the
|
| 73 |
# required LLM head-dimension floor.
|
|
|
|
| 74 |
raw.update(
|
| 75 |
hidden_size=32, head_dim=32, num_attention_heads=1, num_key_value_heads=1,
|
| 76 |
intermediate_size=128, moe_intermediate_size=32, shared_expert_intermediate_size=64,
|
| 77 |
num_experts=64, num_experts_per_tok=6, max_position_embeddings=4096,
|
|
|
|
|
|
|
|
|
|
| 78 |
)
|
| 79 |
(folder / "config.json").write_text(json.dumps(raw, indent=2) + "\n", encoding="utf-8")
|
| 80 |
generator = torch.Generator(device="cpu").manual_seed(42)
|
|
|
|
| 13 |
| Loading implementation | source SGLang plugin | same source plugin wheel |
|
| 14 |
| Stored precision | BF16 | BF16 |
|
| 15 |
| Vocabulary / embedding and LM-head ties | 150,144 / untied | same |
|
| 16 |
+
| Layers / dense prefix | 49 / layers 0-1 | 8 / layers 0-1 |
|
| 17 |
+
| Attention cadence / RoPE-free layers | 49-layer source schedule | first 8 source layers |
|
| 18 |
| Hidden size / Q and KV heads / head dimension | 2,048 / 32 and 4 / 128 | 32 / 1 and 1 / 32 |
|
| 19 |
| Routed experts / active experts | 256 / 6 | 64 / 6 |
|
| 20 |
| MoE / shared-expert widths | 512 / 1,024 | 32 / 64 |
|
|
|
|
| 27 |
|
| 28 |
| File path | Size |
|
| 29 |
|------|------|
|
| 30 |
+
| model.safetensors | 21.9MB |
|
| 31 |
|
| 32 |
|
| 33 |
### Example usage:
|
|
|
|
| 71 |
# The fixed 150,144-row untied embedding and LM head make hidden_size=32 the
|
| 72 |
# smallest practical BF16 geometry; one 32-wide attention head retains the
|
| 73 |
# required LLM head-dimension floor.
|
| 74 |
+
layers = 8
|
| 75 |
raw.update(
|
| 76 |
hidden_size=32, head_dim=32, num_attention_heads=1, num_key_value_heads=1,
|
| 77 |
intermediate_size=128, moe_intermediate_size=32, shared_expert_intermediate_size=64,
|
| 78 |
num_experts=64, num_experts_per_tok=6, max_position_embeddings=4096,
|
| 79 |
+
num_hidden_layers=layers,
|
| 80 |
+
layer_types=raw["layer_types"][:layers],
|
| 81 |
+
no_rope_layer_indices=[index for index in raw["no_rope_layer_indices"] if index < layers],
|
| 82 |
)
|
| 83 |
(folder / "config.json").write_text(json.dumps(raw, indent=2) + "\n", encoding="utf-8")
|
| 84 |
generator = torch.Generator(device="cpu").manual_seed(42)
|
config.json
CHANGED
|
@@ -26,48 +26,7 @@
|
|
| 26 |
"sliding_attention",
|
| 27 |
"full_attention",
|
| 28 |
"sliding_attention",
|
| 29 |
-
"sliding_attention"
|
| 30 |
-
"sliding_attention",
|
| 31 |
-
"full_attention",
|
| 32 |
-
"sliding_attention",
|
| 33 |
-
"sliding_attention",
|
| 34 |
-
"sliding_attention",
|
| 35 |
-
"full_attention",
|
| 36 |
-
"sliding_attention",
|
| 37 |
-
"sliding_attention",
|
| 38 |
-
"sliding_attention",
|
| 39 |
-
"full_attention",
|
| 40 |
-
"sliding_attention",
|
| 41 |
-
"sliding_attention",
|
| 42 |
-
"sliding_attention",
|
| 43 |
-
"full_attention",
|
| 44 |
-
"sliding_attention",
|
| 45 |
-
"sliding_attention",
|
| 46 |
-
"sliding_attention",
|
| 47 |
-
"full_attention",
|
| 48 |
-
"sliding_attention",
|
| 49 |
-
"sliding_attention",
|
| 50 |
-
"sliding_attention",
|
| 51 |
-
"full_attention",
|
| 52 |
-
"sliding_attention",
|
| 53 |
-
"sliding_attention",
|
| 54 |
-
"sliding_attention",
|
| 55 |
-
"full_attention",
|
| 56 |
-
"sliding_attention",
|
| 57 |
-
"sliding_attention",
|
| 58 |
-
"sliding_attention",
|
| 59 |
-
"full_attention",
|
| 60 |
-
"sliding_attention",
|
| 61 |
-
"sliding_attention",
|
| 62 |
-
"sliding_attention",
|
| 63 |
-
"full_attention",
|
| 64 |
-
"sliding_attention",
|
| 65 |
-
"sliding_attention",
|
| 66 |
-
"sliding_attention",
|
| 67 |
-
"full_attention",
|
| 68 |
-
"sliding_attention",
|
| 69 |
-
"sliding_attention",
|
| 70 |
-
"full_attention"
|
| 71 |
],
|
| 72 |
"max_position_embeddings": 4096,
|
| 73 |
"mlp_only_layers": [
|
|
@@ -79,23 +38,12 @@
|
|
| 79 |
"num_attention_heads": 1,
|
| 80 |
"num_experts": 64,
|
| 81 |
"num_experts_per_tok": 6,
|
| 82 |
-
"num_hidden_layers":
|
| 83 |
"num_key_value_heads": 1,
|
| 84 |
"no_rope_layer_indices": [
|
| 85 |
0,
|
| 86 |
1,
|
| 87 |
-
5
|
| 88 |
-
9,
|
| 89 |
-
13,
|
| 90 |
-
17,
|
| 91 |
-
21,
|
| 92 |
-
25,
|
| 93 |
-
29,
|
| 94 |
-
33,
|
| 95 |
-
37,
|
| 96 |
-
41,
|
| 97 |
-
45,
|
| 98 |
-
48
|
| 99 |
],
|
| 100 |
"output_router_logits": false,
|
| 101 |
"pad_token_id": 150001,
|
|
|
|
| 26 |
"sliding_attention",
|
| 27 |
"full_attention",
|
| 28 |
"sliding_attention",
|
| 29 |
+
"sliding_attention"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 30 |
],
|
| 31 |
"max_position_embeddings": 4096,
|
| 32 |
"mlp_only_layers": [
|
|
|
|
| 38 |
"num_attention_heads": 1,
|
| 39 |
"num_experts": 64,
|
| 40 |
"num_experts_per_tok": 6,
|
| 41 |
+
"num_hidden_layers": 8,
|
| 42 |
"num_key_value_heads": 1,
|
| 43 |
"no_rope_layer_indices": [
|
| 44 |
0,
|
| 45 |
1,
|
| 46 |
+
5
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 47 |
],
|
| 48 |
"output_router_logits": false,
|
| 49 |
"pad_token_id": 150001,
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e38ba949d64adbabd3772099563273468f7f0a428a8274b9ae15b0f18ec03faa
|
| 3 |
+
size 21938168
|