patrickvonplaten commited on
Commit
1a8c226
1 Parent(s): 49d34bc

add tf model

Browse files
Files changed (2) hide show
  1. config.json +7 -1
  2. tf_model.h5 +3 -0
config.json CHANGED
@@ -1,4 +1,5 @@
1
  {
 
2
  "_num_labels": 3,
3
  "activation_dropout": 0.0,
4
  "activation_function": "swish",
@@ -15,6 +16,7 @@
15
  ],
16
  "bos_token_id": 0,
17
  "classif_dropout": 0.0,
 
18
  "d_model": 512,
19
  "decoder_attention_heads": 8,
20
  "decoder_ffn_dim": 2048,
@@ -27,6 +29,8 @@
27
  "encoder_layerdrop": 0.0,
28
  "encoder_layers": 6,
29
  "eos_token_id": 0,
 
 
30
  "id2label": {
31
  "0": "LABEL_0",
32
  "1": "LABEL_1",
@@ -49,5 +53,7 @@
49
  "pad_token_id": 58100,
50
  "scale_embedding": true,
51
  "static_position_embeddings": true,
 
 
52
  "vocab_size": 58101
53
- }
1
  {
2
+ "_name_or_path": "Helsinki-NLP/opus-mt-en-de",
3
  "_num_labels": 3,
4
  "activation_dropout": 0.0,
5
  "activation_function": "swish",
16
  ],
17
  "bos_token_id": 0,
18
  "classif_dropout": 0.0,
19
+ "classifier_dropout": 0.0,
20
  "d_model": 512,
21
  "decoder_attention_heads": 8,
22
  "decoder_ffn_dim": 2048,
29
  "encoder_layerdrop": 0.0,
30
  "encoder_layers": 6,
31
  "eos_token_id": 0,
32
+ "forced_eos_token_id": 0,
33
+ "gradient_checkpointing": false,
34
  "id2label": {
35
  "0": "LABEL_0",
36
  "1": "LABEL_1",
53
  "pad_token_id": 58100,
54
  "scale_embedding": true,
55
  "static_position_embeddings": true,
56
+ "transformers_version": "4.4.0.dev0",
57
+ "use_cache": true,
58
  "vocab_size": 58101
59
+ }
tf_model.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:69118ee40ce5b461a3a216a1cb3fe0154151d18de13970ef7371b0c165a81dc4
3
+ size 297909280