From d85f52d76cd2d464146fb96b393df57f21854a4f Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Sun, 10 Jan 2021 19:17:25 +0000 Subject: [PATCH] upload tf model --- config.json | 6 +++++- tf_model.h5 | 3 +++ 2 files changed, 8 insertions(+), 1 deletion(-) create mode 100644 tf_model.h5 diff --git a/config.json b/config.json index dca077a..113e78b 100644 --- a/config.json +++ b/config.json @@ -1,4 +1,5 @@ { + "_name_or_path": "./", "activation_dropout": 0.0, "activation_function": "gelu", "add_bias_logits": false, @@ -15,6 +16,7 @@ "decoder_ffn_dim": 5120, "decoder_layerdrop": 0.0, "decoder_layers": 12, + "decoder_start_token_id": 1, "do_blenderbot_90_layernorm": true, "dropout": 0.1, "encoder_attention_heads": 32, @@ -25,6 +27,7 @@ "extra_layer_norm": false, "extra_pos_embeddings": 0, "force_bos_token_to_be_generated": false, + "gradient_checkpointing": false, "id2label": { "0": "LABEL_0", "1": "LABEL_1", @@ -42,7 +45,7 @@ "max_length": 60, "max_position_embeddings": 128, "min_length": 20, - "model_type": "bart", + "model_type": "blenderbot", "no_repeat_ngram_size": 3, "normalize_before": true, "normalize_embedding": false, @@ -51,6 +54,7 @@ "pad_token_id": 0, "scale_embedding": true, "static_position_embeddings": false, + "transformers_version": "4.2.0dev0", "unk_token_id": 3, "use_cache": true, "vocab_size": 8008 diff --git a/tf_model.h5 b/tf_model.h5 new file mode 100644 index 0000000..5ddcb8f --- /dev/null +++ b/tf_model.h5 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0fcd8f7e4b528fae6a97bd32b386ec02dec91f864dffdcdcc1f2d51ea898e793 +size 1459579344