diff --git a/config.json b/config.json index 1b3b118..bd932fc 100644 --- a/config.json +++ b/config.json @@ -1,8 +1,9 @@ { + "_name_or_path": "openai/whisper-tiny", "activation_dropout": 0.0, "activation_function": "gelu", "architectures": [ - "WhisperModel" + "WhisperForConditionalGeneration" ], "attention_dropout": 0.0, "bos_token_id": 50257, @@ -11,11 +12,7 @@ "decoder_ffn_dim": 1536, "decoder_layerdrop": 0.0, "decoder_layers": 4, - "decoder_start_token_id": [ - 50258, - 50259, - 50359 - ], + "decoder_start_token_id": 50257, "dropout": 0.0, "encoder_attention_heads": 6, "encoder_ffn_dim": 1536, diff --git a/pytorch_model.bin b/pytorch_model.bin index 3ec7c8d..12d457f 100644 --- a/pytorch_model.bin +++ b/pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:b0b674e5d559c60a9bfe339edf44c47cbb4b5253c013ab8a02ff1cfbf888fc7a -size 151094003 +oid sha256:9607f98a2b22d9e229ae43c52ecea79dcede9e0c5cfae67e8da6eda86d8aac1d +size 151095027