feat(transformers): Add native Transformers config

#31
Files changed (1) hide show
  1. config.json +8 -8
config.json CHANGED
@@ -22,22 +22,22 @@
22
  "layer_norm_eps": 1e-12,
23
  "layer_norm_type": "layer_norm",
24
  "max_position_embeddings": 8192,
25
- "model_type": "new",
26
  "num_attention_heads": 12,
27
  "num_hidden_layers": 12,
28
  "num_labels": 1,
29
  "pack_qkv": true,
30
  "pad_token_id": 1,
31
  "position_embedding_type": "rope",
32
- "rope_scaling": {
33
- "factor": 8.0,
34
- "type": "ntk"
35
- },
36
- "rope_theta": 20000,
37
  "torch_dtype": "float16",
38
  "transformers_version": "4.39.1",
39
  "type_vocab_size": 1,
40
  "unpad_inputs": false,
41
  "use_memory_efficient_attention": false,
42
- "vocab_size": 250048
43
- }
 
 
 
 
 
 
22
  "layer_norm_eps": 1e-12,
23
  "layer_norm_type": "layer_norm",
24
  "max_position_embeddings": 8192,
25
+ "model_type": "gte",
26
  "num_attention_heads": 12,
27
  "num_hidden_layers": 12,
28
  "num_labels": 1,
29
  "pack_qkv": true,
30
  "pad_token_id": 1,
31
  "position_embedding_type": "rope",
 
 
 
 
 
32
  "torch_dtype": "float16",
33
  "transformers_version": "4.39.1",
34
  "type_vocab_size": 1,
35
  "unpad_inputs": false,
36
  "use_memory_efficient_attention": false,
37
+ "vocab_size": 250048,
38
+ "rope_parameters": {
39
+ "rope_type": "linear",
40
+ "factor": 1.0671404006768237,
41
+ "rope_theta": 160000.0
42
+ }
43
+ }