system HF staff commited on
Commit
7006394
1 Parent(s): 5937b14

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +8 -1
config.json CHANGED
@@ -8,6 +8,11 @@
8
  "MarianMTModel"
9
  ],
10
  "attention_dropout": 0.0,
 
 
 
 
 
11
  "bos_token_id": 0,
12
  "classif_dropout": 0.0,
13
  "d_model": 512,
@@ -15,6 +20,7 @@
15
  "decoder_ffn_dim": 2048,
16
  "decoder_layerdrop": 0.0,
17
  "decoder_layers": 6,
 
18
  "dropout": 0.1,
19
  "encoder_attention_heads": 8,
20
  "encoder_ffn_dim": 2048,
@@ -33,8 +39,9 @@
33
  "LABEL_1": 1,
34
  "LABEL_2": 2
35
  },
 
36
  "max_position_embeddings": 512,
37
- "model_type": "bart",
38
  "normalize_before": false,
39
  "normalize_embedding": false,
40
  "num_beams": 6,
 
8
  "MarianMTModel"
9
  ],
10
  "attention_dropout": 0.0,
11
+ "bad_words_ids": [
12
+ [
13
+ 61898
14
+ ]
15
+ ],
16
  "bos_token_id": 0,
17
  "classif_dropout": 0.0,
18
  "d_model": 512,
 
20
  "decoder_ffn_dim": 2048,
21
  "decoder_layerdrop": 0.0,
22
  "decoder_layers": 6,
23
+ "decoder_start_token_id": 61898,
24
  "dropout": 0.1,
25
  "encoder_attention_heads": 8,
26
  "encoder_ffn_dim": 2048,
 
39
  "LABEL_1": 1,
40
  "LABEL_2": 2
41
  },
42
+ "max_length": 512,
43
  "max_position_embeddings": 512,
44
+ "model_type": "marian",
45
  "normalize_before": false,
46
  "normalize_embedding": false,
47
  "num_beams": 6,