XBFDI commited on
Commit
448c5b3
·
verified ·
1 Parent(s): c49a306

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +9 -1
config.json CHANGED
@@ -1,8 +1,16 @@
1
  {
 
 
 
 
 
 
 
2
  "vocab_size": 64,
3
  "d_model": 192,
4
  "n_layer": 5,
5
  "n_head": 4,
6
  "ctx": 128,
7
- "dropout": 0.15
 
8
  }
 
1
  {
2
+ "architectures": ["EasyFormerLMHeadModel"],
3
+ "model_type": "transformer",
4
+ "auto_map": {
5
+ "AutoConfig": "configuration_easyformer.EasyFormerConfig",
6
+ "AutoModel": "modeling_easyformer.EasyFormerLMHeadModel",
7
+ "AutoModelForCausalLM": "modeling_easyformer.EasyFormerLMHeadModel"
8
+ },
9
  "vocab_size": 64,
10
  "d_model": 192,
11
  "n_layer": 5,
12
  "n_head": 4,
13
  "ctx": 128,
14
+ "dropout": 0.15,
15
+ "torch_dtype": "float32"
16
  }