diff --git a/config.json b/config.json index c0f3a68..3efc49f 100644 --- a/config.json +++ b/config.json @@ -1,6 +1,10 @@ { + "_name_or_path": "openai/whisper-tiny.en", "activation_dropout": 0.0, "activation_function": "gelu", + "architectures": [ + "WhisperForConditionalGeneration" + ], "attention_dropout": 0.0, "bos_token_id": 50257, "d_model": 384, @@ -14,7 +18,7 @@ "encoder_ffn_dim": 1536, "encoder_layerdrop": 0.0, "encoder_layers": 4, - "eos_token_id": 50257, + "eos_token_id": 50256, "feature_size": 1, "init_std": 0.02, "input_channels": 1, @@ -112,6 +116,7 @@ "num_mel_bins": 80, "pad_token_id": 0, "scale_embedding": false, + "torch_dtype": "float32", "transformers_version": "4.23.0.dev0", "use_cache": true, "vocab_size": 51864 diff --git a/pytorch_model.bin b/pytorch_model.bin index c5baa0b..c12df6e 100644 --- a/pytorch_model.bin +++ b/pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:3d3a696a9d83e850686e97184792dd40beb5a06a10febb652f124e9ac32e1f19 -size 151092467 +oid sha256:65400d0bdf90fa9172968cf86ed5cdb1a33fb74944813e37671f4d98141cf02e +size 151095795