From df7df4d2107d3882675bfd5ad50db1aa8dd7d7be Mon Sep 17 00:00:00 2001 From: patil-suraj Date: Mon, 14 Jun 2021 07:34:28 +0000 Subject: [PATCH] add flax model --- .gitattributes | 1 + config.json | 9 +++++++++ flax_model.msgpack | 3 +++ 3 files changed, 13 insertions(+) create mode 100644 flax_model.msgpack diff --git a/.gitattributes b/.gitattributes index dc08351..ae8c63d 100644 --- a/.gitattributes +++ b/.gitattributes @@ -6,3 +6,4 @@ *.tar.gz filter=lfs diff=lfs merge=lfs -text *.ot filter=lfs diff=lfs merge=lfs -text *.onnx filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text diff --git a/config.json b/config.json index 948a13b..dfc96c7 100644 --- a/config.json +++ b/config.json @@ -3,20 +3,27 @@ "activation_dropout": 0.0, "activation_function": "gelu", "add_final_layer_norm": false, + "architectures": [ + "BartForSequenceClassification" + ], "attention_dropout": 0.0, "bos_token_id": 0, "classif_dropout": 0.0, + "classifier_dropout": 0.0, "d_model": 1024, "decoder_attention_heads": 16, "decoder_ffn_dim": 4096, "decoder_layerdrop": 0.0, "decoder_layers": 12, + "decoder_start_token_id": 2, "dropout": 0.1, "encoder_attention_heads": 16, "encoder_ffn_dim": 4096, "encoder_layerdrop": 0.0, "encoder_layers": 12, "eos_token_id": 2, + "forced_eos_token_id": 2, + "gradient_checkpointing": false, "id2label": { "0": "contradiction", "1": "neutral", @@ -36,5 +43,7 @@ "output_past": false, "pad_token_id": 1, "scale_embedding": false, + "transformers_version": "4.7.0.dev0", + "use_cache": true, "vocab_size": 50265 } diff --git a/flax_model.msgpack b/flax_model.msgpack new file mode 100644 index 0000000..1f7b4d2 --- /dev/null +++ b/flax_model.msgpack @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5e07d1ae73ae1c1267fd174a3b21c73b0d77bad288f8ed17fb685f79c419a897 +size 1629394629