klemenk commited on
Commit
edbfcc5
·
verified ·
1 Parent(s): 5814ef1

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +4 -3
config.json CHANGED
@@ -8,11 +8,12 @@
8
  "AutoModelForCausalLM": "modeling_auristream.AuriStream"
9
  },
10
  "vocab_size": 8192,
11
- "n_embd": 768,
12
- "n_layer": 12,
13
- "n_head": 12,
14
  "dropout": 0.0,
15
  "bias": false,
16
  "use_rope": true,
 
17
  "n_pred_steps": 20
18
  }
 
8
  "AutoModelForCausalLM": "modeling_auristream.AuriStream"
9
  },
10
  "vocab_size": 8192,
11
+ "n_embd": 4096,
12
+ "n_layer": 32,
13
+ "n_head": 32,
14
  "dropout": 0.0,
15
  "bias": false,
16
  "use_rope": true,
17
+ "skip_connections": true,
18
  "n_pred_steps": 20
19
  }