patrickvonplaten commited on
Commit
32f2b87
1 Parent(s): 9aba671

corect checkpoint

Browse files
Files changed (2) hide show
  1. config.json +19 -3
  2. pytorch_model.bin +2 -2
config.json CHANGED
@@ -1,15 +1,30 @@
1
  {
 
2
  "_num_labels": 3,
3
  "activation_dropout": 0.0,
4
  "activation_function": "gelu",
5
  "architectures": [
6
- "BartForConditionalGeneration"
7
  ],
8
  "attention_dropout": 0.0,
9
  "attention_probs_dropout_prob": 0.0,
10
- "attention_window": 1024,
 
 
 
 
 
 
 
 
 
 
 
 
 
11
  "bos_token_id": 0,
12
  "classif_dropout": 0.0,
 
13
  "d_model": 1024,
14
  "decoder_attention_heads": 16,
15
  "decoder_ffn_dim": 4096,
@@ -37,10 +52,11 @@
37
  },
38
  "max_decoder_position_embeddings": 1024,
39
  "max_encoder_position_embeddings": 16384,
40
- "model_type": "bart",
41
  "num_hidden_layers": 12,
42
  "output_past": false,
43
  "pad_token_id": 1,
44
  "prefix": " ",
 
45
  "vocab_size": 50265
46
  }
 
1
  {
2
+ "_name_or_path": "./led-large-16384",
3
  "_num_labels": 3,
4
  "activation_dropout": 0.0,
5
  "activation_function": "gelu",
6
  "architectures": [
7
+ "LEDForConditionalGeneration"
8
  ],
9
  "attention_dropout": 0.0,
10
  "attention_probs_dropout_prob": 0.0,
11
+ "attention_window": [
12
+ 1024,
13
+ 1024,
14
+ 1024,
15
+ 1024,
16
+ 1024,
17
+ 1024,
18
+ 1024,
19
+ 1024,
20
+ 1024,
21
+ 1024,
22
+ 1024,
23
+ 1024
24
+ ],
25
  "bos_token_id": 0,
26
  "classif_dropout": 0.0,
27
+ "classifier_dropout": 0.0,
28
  "d_model": 1024,
29
  "decoder_attention_heads": 16,
30
  "decoder_ffn_dim": 4096,
 
52
  },
53
  "max_decoder_position_embeddings": 1024,
54
  "max_encoder_position_embeddings": 16384,
55
+ "model_type": "led",
56
  "num_hidden_layers": 12,
57
  "output_past": false,
58
  "pad_token_id": 1,
59
  "prefix": " ",
60
+ "use_cache": true,
61
  "vocab_size": 50265
62
  }
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d68af582e11d949b625d046bc6a7a7c7e44d1af14a6a2024efad7e485b7480e9
3
- size 1688408119
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7418c41296619d3e5ed5aade6a36fcd86e7faebffa4745a1dda7c4178e923a79
3
+ size 1839652087