Add default translation to config
Browse filesAdd default values because the model wasn't happy that `src_lang` and `tgt_lang` weren't provided
![Screenshot_2023-02-09_23.09.31.png](https://s3.amazonaws.com/moonup/production/uploads/1676017992849-637c850d42bed44b0319aa6e.png)
- config.json +20 -0
config.json
CHANGED
@@ -29,6 +29,26 @@
|
|
29 |
"transformers_version": "4.21.0.dev0",
|
30 |
"use_cache": true,
|
31 |
"vocab_size": 256206,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
32 |
"tokenizer_class": "NllbTokenizer",
|
33 |
"max_length": 200
|
34 |
}
|
|
|
29 |
"transformers_version": "4.21.0.dev0",
|
30 |
"use_cache": true,
|
31 |
"vocab_size": 256206,
|
32 |
+
"task_specific_params": {
|
33 |
+
"translation_en_to_de": {
|
34 |
+
"early_stopping": true,
|
35 |
+
"max_length": 300,
|
36 |
+
"num_beams": 4,
|
37 |
+
"prefix": "translate English to German: "
|
38 |
+
},
|
39 |
+
"translation_en_to_fr": {
|
40 |
+
"early_stopping": true,
|
41 |
+
"max_length": 300,
|
42 |
+
"num_beams": 4,
|
43 |
+
"prefix": "translate English to French: "
|
44 |
+
},
|
45 |
+
"translation_en_to_ro": {
|
46 |
+
"early_stopping": true,
|
47 |
+
"max_length": 300,
|
48 |
+
"num_beams": 4,
|
49 |
+
"prefix": "translate English to Romanian: "
|
50 |
+
}
|
51 |
+
},
|
52 |
"tokenizer_class": "NllbTokenizer",
|
53 |
"max_length": 200
|
54 |
}
|