valhalla commited on
Commit
ed9a6fc
1 Parent(s): 4ae71b2

add flax model

Browse files

Files changed (3) hide show
  1. .gitattributes +1 -0
  2. config.json +5 -0
  3. flax_model.msgpack +3 -0
.gitattributes CHANGED
@@ -6,3 +6,4 @@
6
  *.tar.gz filter=lfs diff=lfs merge=lfs -text
7
  *.ot filter=lfs diff=lfs merge=lfs -text
8
  *.onnx filter=lfs diff=lfs merge=lfs -text
 
6
  *.tar.gz filter=lfs diff=lfs merge=lfs -text
7
  *.ot filter=lfs diff=lfs merge=lfs -text
8
  *.onnx filter=lfs diff=lfs merge=lfs -text
9
+ *.msgpack filter=lfs diff=lfs merge=lfs -text
config.json CHANGED
@@ -10,6 +10,7 @@
10
  "attention_dropout": 0.0,
11
  "bos_token_id": 0,
12
  "classif_dropout": 0.0,
 
13
  "d_model": 24,
14
  "decoder_attention_heads": 2,
15
  "decoder_ffn_dim": 16,
@@ -22,6 +23,8 @@
22
  "encoder_layerdrop": 0.0,
23
  "encoder_layers": 2,
24
  "eos_token_id": 2,
 
 
25
  "id2label": {
26
  "0": "LABEL_0",
27
  "1": "LABEL_1",
@@ -54,5 +57,7 @@
54
  "num_beams": 4
55
  }
56
  },
 
 
57
  "vocab_size": 50265
58
  }
10
  "attention_dropout": 0.0,
11
  "bos_token_id": 0,
12
  "classif_dropout": 0.0,
13
+ "classifier_dropout": 0.0,
14
  "d_model": 24,
15
  "decoder_attention_heads": 2,
16
  "decoder_ffn_dim": 16,
23
  "encoder_layerdrop": 0.0,
24
  "encoder_layers": 2,
25
  "eos_token_id": 2,
26
+ "forced_eos_token_id": 2,
27
+ "gradient_checkpointing": false,
28
  "id2label": {
29
  "0": "LABEL_0",
30
  "1": "LABEL_1",
57
  "num_beams": 4
58
  }
59
  },
60
+ "transformers_version": "4.7.0.dev0",
61
+ "use_cache": true,
62
  "vocab_size": 50265
63
  }
flax_model.msgpack ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:11d261660860c4905d08cbffcec0abd311d35d477cde1158594b6a35b9042b41
3
+ size 5275378