system HF staff commited on
Commit
c32744d
1 Parent(s): b4d372a

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +24 -4
config.json CHANGED
@@ -1,5 +1,7 @@
1
  {
2
- "architectures": ["T5ForConditionalGeneration"],
 
 
3
  "d_ff": 3072,
4
  "d_kv": 64,
5
  "d_model": 768,
@@ -8,7 +10,7 @@
8
  "eos_token_id": 1,
9
  "initializer_factor": 1.0,
10
  "is_encoder_decoder": true,
11
- "layer_norm_epsilon": 1e-6,
12
  "model_type": "t5",
13
  "n_positions": 512,
14
  "num_heads": 12,
@@ -16,7 +18,6 @@
16
  "output_past": true,
17
  "pad_token_id": 0,
18
  "relative_attention_num_buckets": 32,
19
- "vocab_size": 32128,
20
  "task_specific_params": {
21
  "summarization": {
22
  "early_stopping": true,
@@ -26,6 +27,25 @@
26
  "no_repeat_ngram_size": 3,
27
  "num_beams": 4,
28
  "prefix": "summarize: "
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
29
  }
30
- }
 
31
  }
 
1
  {
2
+ "architectures": [
3
+ "T5ForConditionalGeneration"
4
+ ],
5
  "d_ff": 3072,
6
  "d_kv": 64,
7
  "d_model": 768,
 
10
  "eos_token_id": 1,
11
  "initializer_factor": 1.0,
12
  "is_encoder_decoder": true,
13
+ "layer_norm_epsilon": 1e-06,
14
  "model_type": "t5",
15
  "n_positions": 512,
16
  "num_heads": 12,
 
18
  "output_past": true,
19
  "pad_token_id": 0,
20
  "relative_attention_num_buckets": 32,
 
21
  "task_specific_params": {
22
  "summarization": {
23
  "early_stopping": true,
 
27
  "no_repeat_ngram_size": 3,
28
  "num_beams": 4,
29
  "prefix": "summarize: "
30
+ },
31
+ "translation_en_to_de": {
32
+ "early_stopping": true,
33
+ "max_length": 300,
34
+ "num_beams": 4,
35
+ "prefix": "translate English to German: "
36
+ },
37
+ "translation_en_to_fr": {
38
+ "early_stopping": true,
39
+ "max_length": 300,
40
+ "num_beams": 4,
41
+ "prefix": "translate English to French: "
42
+ },
43
+ "translation_en_to_ro": {
44
+ "early_stopping": true,
45
+ "max_length": 300,
46
+ "num_beams": 4,
47
+ "prefix": "translate English to Romanian: "
48
  }
49
+ },
50
+ "vocab_size": 32128
51
  }