Scorched2 commited on
Commit
3a62af9
verified
1 Parent(s): 461f4de

Upload folder using huggingface_hub

Browse files
README.md ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ tags: [hrm, moe, hierarchical-reasoning, custom-architecture]
4
+ ---
5
+
6
+ # Hierarchical Reasoning Model (HRM)
7
+
8
+ Custom MoE language model with 3-level hierarchical reasoning and DeepSeek-V3 memory strategies.
9
+
10
+ **Architecture:** 3 levels 路 16 experts (4 active) 路 MLA attention 路 Hierarchical memory
11
+ **Parameters:** ~350M total, ~45M active per token
12
+ **Trained on:** OpenHermes-2.5
13
+
14
+ ## Usage
15
+
16
+ ```python
17
+ from transformers import AutoModelForCausalLM, AutoTokenizer
18
+
19
+ model = AutoModelForCausalLM.from_pretrained("Scorched2/shader-v2", trust_remote_code=True)
20
+ tokenizer = AutoTokenizer.from_pretrained("Scorched2/shader-v2")
21
+
22
+ inputs = tokenizer("### Instruction:\nExplain AI.\n\n### Response:\n", return_tensors="pt")
23
+ out = model.generate(**inputs, max_new_tokens=200)
24
+ print(tokenizer.decode(out[0]))
25
+ ```
config.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "HRMForCausalLM"
4
+ ],
5
+ "bos_token_id": 50256,
6
+ "d_ff": 1536,
7
+ "d_model": 768,
8
+ "dropout": 0.0,
9
+ "dtype": "float32",
10
+ "eos_token_id": 50256,
11
+ "episodic_mem_slots": 64,
12
+ "head_dim": 64,
13
+ "kv_lora_rank": 192,
14
+ "layers_per_level": [
15
+ 3,
16
+ 3,
17
+ 3
18
+ ],
19
+ "max_seq_len": 256,
20
+ "model_type": "hrm",
21
+ "moe_d_ff": 512,
22
+ "mtp_heads": 2,
23
+ "n_active_experts": 4,
24
+ "n_experts": 16,
25
+ "n_heads": 12,
26
+ "n_levels": 3,
27
+ "n_shared_experts": 1,
28
+ "pad_token_id": 0,
29
+ "q_lora_rank": 256,
30
+ "rope_head_dim": 32,
31
+ "semantic_mem_dim": 768,
32
+ "semantic_mem_size": 512,
33
+ "tie_embeddings": true,
34
+ "transformers_version": "5.7.0",
35
+ "vocab_size": 50257,
36
+ "working_mem_slots": 16
37
+ }
generation_config.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 50256,
4
+ "eos_token_id": 50256,
5
+ "output_attentions": false,
6
+ "output_hidden_states": false,
7
+ "pad_token_id": 0,
8
+ "transformers_version": "5.7.0"
9
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bd7854934776f17859694a4f26ae0bcf329f9c6ccf3f932ee16ae6bb45878101
3
+ size 1301867064
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "backend": "tokenizers",
4
+ "bos_token": "<|endoftext|>",
5
+ "eos_token": "<|endoftext|>",
6
+ "errors": "replace",
7
+ "is_local": false,
8
+ "local_files_only": false,
9
+ "model_max_length": 1024,
10
+ "pad_token": "<|endoftext|>",
11
+ "tokenizer_class": "GPT2Tokenizer",
12
+ "unk_token": "<|endoftext|>"
13
+ }