From d902ab9a804fd74a84c140adc2fdb004472c7564 Mon Sep 17 00:00:00 2001 From: Arthur Zucker Date: Tue, 27 Sep 2022 07:02:42 +0000 Subject: [PATCH] Upload WhisperForConditionalGeneration --- config.json | 9 +++------ pytorch_model.bin | 4 ++-- 2 files changed, 5 insertions(+), 8 deletions(-) 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