sheonhan commited on
Commit
6e750da
1 Parent(s): 061c198

Add default translation to config

Browse files

Add 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)

Files changed (1) hide show
  1. 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
  }