system HF staff commited on
Commit
94a6bec
1 Parent(s): e1df70f

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +4 -2
config.json CHANGED
@@ -36,10 +36,12 @@
36
  "max_position_embeddings": 1024,
37
  "model_type": "mbart",
38
  "normalize_before": true,
39
- "variant": "prelayernorm",
40
  "normalize_embedding": true,
 
 
 
41
  "num_beams": 5,
42
- "max_length": 1000,
43
  "num_hidden_layers": 12,
44
  "output_past": true,
45
  "pad_token_id": 1,
 
36
  "max_position_embeddings": 1024,
37
  "model_type": "mbart",
38
  "normalize_before": true,
 
39
  "normalize_embedding": true,
40
+ "task_specific_params": {
41
+ "translation_en_to_ro": { "decoder_start_token_id": 250020}
42
+ },
43
  "num_beams": 5,
44
+ "max_length": 1024,
45
  "num_hidden_layers": 12,
46
  "output_past": true,
47
  "pad_token_id": 1,