openai
/
whisper-large-v2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add Flax weights

This commit is contained in:
sanchit-gandhi 2023-02-23 17:15:19 +01:00
parent bd0efe4d58
commit 5ff64998d3
3 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
config.json
View File

@ -136,7 +136,7 @@
50362 50362
], ],
"torch_dtype": "float32", "torch_dtype": "float32",
"transformers_version": "4.26.0.dev0", "transformers_version": "4.27.0.dev0",
"use_cache": true, "use_cache": true,
"vocab_size": 51865 "vocab_size": 51865
} }

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

Binary file not shown.

4
generation_config.json
View File

@ -3,7 +3,6 @@
220, 220,
50257 50257
], ],
"return_timestamps":false,
"bos_token_id": 50257, "bos_token_id": 50257,
"decoder_start_token_id": 50258, "decoder_start_token_id": 50258,
"eos_token_id": 50257, "eos_token_id": 50257,
@ -123,6 +122,7 @@
"max_length": 448, "max_length": 448,
"no_timestamps_token_id": 50363, "no_timestamps_token_id": 50363,
"pad_token_id": 50256, "pad_token_id": 50256,
"return_timestamps": false,
"suppress_tokens": [ "suppress_tokens": [
1, 1,
2, 2,
@ -215,5 +215,5 @@
"transcribe": 50359, "transcribe": 50359,
"translate": 50358 "translate": 50358
}, },
"transformers_version": "4.26.0.dev0" "transformers_version": "4.27.0.dev0"
} }