tlkh commited on
Commit
bd61bef
1 Parent(s): 979ea97

updated config

Browse files
Files changed (1) hide show
  1. config.json +3 -24
config.json CHANGED
@@ -23,32 +23,11 @@
23
  "pad_token_id": 0,
24
  "relative_attention_num_buckets": 32,
25
  "task_specific_params": {
26
- "summarization": {
27
  "early_stopping": true,
28
- "length_penalty": 2.0,
29
- "max_length": 200,
30
- "min_length": 30,
31
- "no_repeat_ngram_size": 3,
32
  "num_beams": 4,
33
- "prefix": "summarize: "
34
- },
35
- "translation_en_to_de": {
36
- "early_stopping": true,
37
- "max_length": 300,
38
- "num_beams": 4,
39
- "prefix": "translate English to German: "
40
- },
41
- "translation_en_to_fr": {
42
- "early_stopping": true,
43
- "max_length": 300,
44
- "num_beams": 4,
45
- "prefix": "translate English to French: "
46
- },
47
- "translation_en_to_ro": {
48
- "early_stopping": true,
49
- "max_length": 300,
50
- "num_beams": 4,
51
- "prefix": "translate English to Romanian: "
52
  }
53
  },
54
  "torch_dtype": "float32",
 
23
  "pad_token_id": 0,
24
  "relative_attention_num_buckets": 32,
25
  "task_specific_params": {
26
+ "translation": {
27
  "early_stopping": true,
28
+ "max_length": 64,
 
 
 
29
  "num_beams": 4,
30
+ "prefix": ""
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
31
  }
32
  },
33
  "torch_dtype": "float32",