Add Flax weights

This commit is contained in:
sanchit-gandhi 2023-02-23 16:31:39 +01:00
parent 74d3cae0a5
commit 1f7ce4a468
3 changed files with 6 additions and 4 deletions

View File

@ -130,7 +130,7 @@
50361 50361
], ],
"torch_dtype": "float32", "torch_dtype": "float32",
"transformers_version": "4.24.0.dev0", "transformers_version": "4.27.0.dev0",
"use_cache": true, "use_cache": true,
"vocab_size": 51864 "vocab_size": 51864
} }

BIN
flax_model.msgpack (Stored with Git LFS) Normal file

Binary file not shown.

View File

@ -1,10 +1,8 @@
{ {
"_from_model_config": false,
"begin_suppress_tokens": [ "begin_suppress_tokens": [
220, 220,
50256 50256
], ],
"return_timestamps":false,
"bos_token_id": 50257, "bos_token_id": 50257,
"decoder_start_token_id": 50257, "decoder_start_token_id": 50257,
"eos_token_id": 50256, "eos_token_id": 50256,
@ -19,6 +17,7 @@
"max_length": 448, "max_length": 448,
"no_timestamps_token_id": 50362, "no_timestamps_token_id": 50362,
"pad_token_id": 50256, "pad_token_id": 50256,
"return_timestamps": false,
"suppress_tokens": [ "suppress_tokens": [
1, 1,
2, 2,
@ -109,5 +108,5 @@
50360, 50360,
50361 50361
], ],
"transformers_version": "4.26.0.dev0" "transformers_version": "4.27.0.dev0"
} }