Girinath11 commited on
Commit
f3f49c3
·
verified ·
1 Parent(s): 03543ee

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +12 -21
config.json CHANGED
@@ -1,37 +1,28 @@
1
  {
2
- "architectures": [
3
- "RecursiveLanguageModel"
4
- ],
5
  "auto_map": {
6
  "AutoConfig": "mixture_of_recursion.RecursiveLanguageModelConfig",
7
- "AutoModel": "mixture_of_recursion.RecursiveLanguageModel",
8
  "AutoModelForCausalLM": "mixture_of_recursion.RecursiveLanguageModel"
9
  },
10
  "model_type": "recursive_language_model",
11
- "transformers_version": "4.35.0",
12
-
13
- "vocab_size": 50259,
14
- "embedding_dim": 560,
15
- "num_layers": 8,
16
- "num_attention_heads": 8,
17
  "max_recursion_steps": 5,
18
  "max_position_embeddings": 512,
19
  "hidden_dropout_prob": 0.1,
20
  "attention_dropout_prob": 0.1,
21
- "intermediate_size": 2240,
22
- "layer_norm_eps": 1e-5,
23
-
24
- "pad_token_id": 50256,
25
  "bos_token_id": 50256,
26
  "eos_token_id": 50256,
27
-
28
  "simple_recursion_steps": 1,
29
  "medium_recursion_steps": 3,
30
  "complex_recursion_steps": 5,
31
-
32
- "confidence_threshold": 0.8,
33
- "use_adaptive_stopping": true,
34
  "initializer_range": 0.02,
35
-
36
- "tie_word_embeddings": false
37
- }
 
1
  {
2
+ "architectures": ["RecursiveLanguageModel"],
 
 
3
  "auto_map": {
4
  "AutoConfig": "mixture_of_recursion.RecursiveLanguageModelConfig",
 
5
  "AutoModelForCausalLM": "mixture_of_recursion.RecursiveLanguageModel"
6
  },
7
  "model_type": "recursive_language_model",
8
+ "transformers_version": "5.2.0",
9
+ "vocab_size": 50260,
10
+ "embedding_dim": 768,
11
+ "num_layers": 16,
12
+ "num_attention_heads": 12,
 
13
  "max_recursion_steps": 5,
14
  "max_position_embeddings": 512,
15
  "hidden_dropout_prob": 0.1,
16
  "attention_dropout_prob": 0.1,
17
+ "intermediate_size": 3072,
18
+ "layer_norm_eps": 1e-05,
19
+ "pad_token_id": 50257,
 
20
  "bos_token_id": 50256,
21
  "eos_token_id": 50256,
 
22
  "simple_recursion_steps": 1,
23
  "medium_recursion_steps": 3,
24
  "complex_recursion_steps": 5,
 
 
 
25
  "initializer_range": 0.02,
26
+ "tie_word_embeddings": false,
27
+ "dtype": "float32"
28
+ }