Sandroeth commited on
Commit
be45155
·
verified ·
1 Parent(s): 0e1c37c

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +11 -10
config.json CHANGED
@@ -1,10 +1,9 @@
1
  {
2
- "architectures": ["CaliForCausalLM"],
3
  "model_type": "cali",
4
-
5
  "auto_map": {
6
- "AutoConfig": "configuration_cali.CaliConfig",
7
- "AutoModelForCausalLM": "modeling_cali.CaliForCausalLM"
8
  },
9
  "vocab_size": 32000,
10
  "hidden_dim": 768,
@@ -13,14 +12,16 @@
13
  "num_kv_heads": 1,
14
  "head_dim": 192,
15
  "seq_len": 1024,
16
- "max_position_embeddings": 1024,
17
  "ffn_multiplier": 3.0,
18
- "ffn_dim": 2304,
19
  "tie_embeddings": true,
20
  "rope_theta": 10000.0,
21
- "bos_token_id": 2,
22
- "eos_token_id": 3,
23
- "pad_token_id": 0,
24
- "torch_dtype": "float16",
 
 
 
25
  "transformers_version": "4.40.0"
26
  }
 
1
  {
 
2
  "model_type": "cali",
3
+ "architectures": ["CALIForCausalLM"],
4
  "auto_map": {
5
+ "AutoConfig": "configuration_cali.CALIConfig",
6
+ "AutoModelForCausalLM": "modeling_cali.CALIForCausalLM"
7
  },
8
  "vocab_size": 32000,
9
  "hidden_dim": 768,
 
12
  "num_kv_heads": 1,
13
  "head_dim": 192,
14
  "seq_len": 1024,
 
15
  "ffn_multiplier": 3.0,
16
+ "dropout": 0.0,
17
  "tie_embeddings": true,
18
  "rope_theta": 10000.0,
19
+ "rms_norm_eps": 1e-6,
20
+ "initializer_range": 0.02,
21
+ "use_cache": true,
22
+ "pad_token_id": null,
23
+ "bos_token_id": 1,
24
+ "eos_token_id": 2,
25
+ "torch_dtype": "float32",
26
  "transformers_version": "4.40.0"
27
  }