joerowell commited on
Commit
12f3ade
·
verified ·
1 Parent(s): 91dc15b

tiny per-element Laguna test fixture

Browse files
Files changed (3) hide show
  1. config.json +62 -0
  2. generation_config.json +7 -0
  3. model.safetensors +3 -0
config.json ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "LagunaForCausalLM"
4
+ ],
5
+ "attention_bias": false,
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": null,
8
+ "dtype": "float32",
9
+ "eos_token_id": null,
10
+ "gating": "per-element",
11
+ "head_dim": 16,
12
+ "hidden_act": "silu",
13
+ "hidden_size": 64,
14
+ "initializer_range": 0.02,
15
+ "intermediate_size": 128,
16
+ "layer_types": [
17
+ "full_attention",
18
+ "full_attention"
19
+ ],
20
+ "max_position_embeddings": 128,
21
+ "mlp_layer_types": [
22
+ "dense",
23
+ "sparse"
24
+ ],
25
+ "model_type": "laguna",
26
+ "moe_apply_router_weight_on_input": false,
27
+ "moe_intermediate_size": 32,
28
+ "moe_routed_scaling_factor": 1.0,
29
+ "moe_router_logit_softcapping": 0.0,
30
+ "num_attention_heads": 8,
31
+ "num_attention_heads_per_layer": [
32
+ 8,
33
+ 8
34
+ ],
35
+ "num_experts": 8,
36
+ "num_experts_per_tok": 2,
37
+ "num_hidden_layers": 2,
38
+ "num_key_value_heads": 2,
39
+ "output_router_logits": false,
40
+ "pad_token_id": null,
41
+ "rms_norm_eps": 1e-06,
42
+ "rope_parameters": {
43
+ "full_attention": {
44
+ "partial_rotary_factor": 0.5,
45
+ "rope_theta": 500000.0,
46
+ "rope_type": "default"
47
+ },
48
+ "rope_type": "default",
49
+ "sliding_attention": {
50
+ "partial_rotary_factor": 1.0,
51
+ "rope_theta": 10000.0,
52
+ "rope_type": "default"
53
+ }
54
+ },
55
+ "router_aux_loss_coef": 0.001,
56
+ "shared_expert_intermediate_size": 32,
57
+ "sliding_window": 8,
58
+ "tie_word_embeddings": false,
59
+ "transformers_version": "5.13.0.dev0",
60
+ "use_cache": true,
61
+ "vocab_size": 256
62
+ }
generation_config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "output_attentions": false,
4
+ "output_hidden_states": false,
5
+ "transformers_version": "5.13.0.dev0",
6
+ "use_cache": true
7
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c1f03bda83dbc48f7fe1e3cce9d43344c6f1056e5d1e050dabd279eb03861c44
3
+ size 689208