From 7676c0925b50115a783b7368d2923a4d3ec87eb4 Mon Sep 17 00:00:00 2001 From: patil-suraj Date: Mon, 14 Jun 2021 07:28:15 +0000 Subject: [PATCH] add flax model --- .gitattributes | 1 + config.json | 7 +++++++ flax_model.msgpack | 3 +++ 3 files changed, 11 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 918613c..2d29716 100644 --- a/config.json +++ b/config.json @@ -3,9 +3,13 @@ "activation_dropout": 0.0, "activation_function": "gelu", "add_final_layer_norm": false, + "architectures": [ + "BartForConditionalGeneration" + ], "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, @@ -22,6 +26,7 @@ "force_bos_token_to_be_generated": true, "forced_bos_token_id": 0, "forced_eos_token_id": 2, + "gradient_checkpointing": false, "id2label": { "0": "LABEL_0", "1": "LABEL_1", @@ -57,5 +62,7 @@ "num_beams": 4 } }, + "transformers_version": "4.7.0.dev0", + "use_cache": true, "vocab_size": 50264 } diff --git a/flax_model.msgpack b/flax_model.msgpack new file mode 100644 index 0000000..827b655 --- /dev/null +++ b/flax_model.msgpack @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:dda1068890b2e54a98369501ceebbed19edfe184a45581c1ecd228e3f224a4bf +size 1625380786