Everlyn commited on
Commit
9f4409a
·
1 Parent(s): 0d0fcfb

Upload model

Browse files
Files changed (1) hide show
  1. config.json +5 -5
config.json CHANGED
@@ -8,22 +8,22 @@
8
  "bos_token_id": 0,
9
  "classifier_dropout": 0.0,
10
  "d_model": 384,
11
- "decoder_attention_heads": 6,
12
  "decoder_ffn_dim": 2048,
13
  "decoder_layerdrop": 0.0,
14
- "decoder_layers": 4,
15
  "dropout": 0.1,
16
- "encoder_attention_heads": 6,
17
  "encoder_ffn_dim": 2048,
18
  "encoder_layerdrop": 0.0,
19
- "encoder_layers": 4,
20
  "eos_token_id": 2,
21
  "forced_eos_token_id": 2,
22
  "init_std": 0.02,
23
  "is_encoder_decoder": true,
24
  "max_position_embeddings": 384,
25
  "model_type": "mbart",
26
- "num_hidden_layers": 4,
27
  "pad_token_id": 1,
28
  "scale_embedding": false,
29
  "transformers_version": "4.26.0",
 
8
  "bos_token_id": 0,
9
  "classifier_dropout": 0.0,
10
  "d_model": 384,
11
+ "decoder_attention_heads": 4,
12
  "decoder_ffn_dim": 2048,
13
  "decoder_layerdrop": 0.0,
14
+ "decoder_layers": 3,
15
  "dropout": 0.1,
16
+ "encoder_attention_heads": 4,
17
  "encoder_ffn_dim": 2048,
18
  "encoder_layerdrop": 0.0,
19
+ "encoder_layers": 3,
20
  "eos_token_id": 2,
21
  "forced_eos_token_id": 2,
22
  "init_std": 0.02,
23
  "is_encoder_decoder": true,
24
  "max_position_embeddings": 384,
25
  "model_type": "mbart",
26
+ "num_hidden_layers": 3,
27
  "pad_token_id": 1,
28
  "scale_embedding": false,
29
  "transformers_version": "4.26.0",