Sandroeth commited on
Commit
6ab1219
·
verified ·
1 Parent(s): 61ace24

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +6 -4
config.json CHANGED
@@ -1,8 +1,6 @@
1
  {
 
2
  "model_type": "cali",
3
- "architectures": [
4
- "CaliForCausalLM"
5
- ],
6
  "auto_map": {
7
  "AutoConfig": "configuration_cali.CaliConfig",
8
  "AutoModelForCausalLM": "modeling_cali.CaliForCausalLM"
@@ -13,6 +11,10 @@
13
  "num_heads": 4,
14
  "num_kv_heads": 1,
15
  "head_dim": 192,
 
16
  "ffn_multiplier": 3.0,
17
- "tie_embeddings": true
 
 
 
18
  }
 
1
  {
2
+ "architectures": ["CaliForCausalLM"],
3
  "model_type": "cali",
 
 
 
4
  "auto_map": {
5
  "AutoConfig": "configuration_cali.CaliConfig",
6
  "AutoModelForCausalLM": "modeling_cali.CaliForCausalLM"
 
11
  "num_heads": 4,
12
  "num_kv_heads": 1,
13
  "head_dim": 192,
14
+ "seq_len": 1024,
15
  "ffn_multiplier": 3.0,
16
+ "tie_embeddings": true,
17
+ "rope_theta": 10000.0,
18
+ "torch_dtype": "float16",
19
+ "transformers_version": "4.40.0"
20
  }