From 1f7ce4a468ac010a9736e54fdbab70cac0d5ea72 Mon Sep 17 00:00:00 2001 From: sanchit-gandhi Date: Thu, 23 Feb 2023 16:31:39 +0100 Subject: [PATCH] Add Flax weights --- config.json | 2 +- flax_model.msgpack | 3 +++ generation_config.json | 5 ++--- 3 files changed, 6 insertions(+), 4 deletions(-) create mode 100644 flax_model.msgpack diff --git a/config.json b/config.json index 18c737b..2fea1ec 100644 --- a/config.json +++ b/config.json @@ -130,7 +130,7 @@ 50361 ], "torch_dtype": "float32", - "transformers_version": "4.24.0.dev0", + "transformers_version": "4.27.0.dev0", "use_cache": true, "vocab_size": 51864 } diff --git a/flax_model.msgpack b/flax_model.msgpack new file mode 100644 index 0000000..087799e --- /dev/null +++ b/flax_model.msgpack @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d0c9b1bfca788c75ea8e557b809b0829341ef835efd9b4ffefbd326b09b7eda2 +size 151047055 diff --git a/generation_config.json b/generation_config.json index 03a7c66..95d6e11 100644 --- a/generation_config.json +++ b/generation_config.json @@ -1,10 +1,8 @@ { - "_from_model_config": false, "begin_suppress_tokens": [ 220, 50256 ], - "return_timestamps":false, "bos_token_id": 50257, "decoder_start_token_id": 50257, "eos_token_id": 50256, @@ -19,6 +17,7 @@ "max_length": 448, "no_timestamps_token_id": 50362, "pad_token_id": 50256, + "return_timestamps": false, "suppress_tokens": [ 1, 2, @@ -109,5 +108,5 @@ 50360, 50361 ], - "transformers_version": "4.26.0.dev0" + "transformers_version": "4.27.0.dev0" }