chopey commited on
Commit
2d6831b
1 Parent(s): c0a35bb

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +5 -4
config.json CHANGED
@@ -14,14 +14,15 @@
14
  "is_encoder_decoder": true,
15
  "layer_norm_epsilon": 1e-06,
16
  "model_type": "mt5",
17
- "num_decoder_layers": 12,
18
- "num_heads": 12,
19
- "num_layers": 12,
20
- "output_past": true,
21
  "pad_token_id": 0,
22
  "relative_attention_num_buckets": 32,
23
  "tie_word_embeddings": false,
24
  "tokenizer_class": "T5Tokenizer",
 
 
25
  "use_cache": true,
26
  "vocab_size": 250112
27
  }
 
14
  "is_encoder_decoder": true,
15
  "layer_norm_epsilon": 1e-06,
16
  "model_type": "mt5",
17
+ "num_decoder_layers": 8,
18
+ "num_heads": 6,
19
+ "num_layers": 8,
 
20
  "pad_token_id": 0,
21
  "relative_attention_num_buckets": 32,
22
  "tie_word_embeddings": false,
23
  "tokenizer_class": "T5Tokenizer",
24
+ "torch_dtype": "float32",
25
+ "transformers_version": "4.12.5",
26
  "use_cache": true,
27
  "vocab_size": 250112
28
  }