diff --git a/config.json b/config.json index bd932fc..9e9121d 100644 --- a/config.json +++ b/config.json @@ -1,10 +1,6 @@ { - "_name_or_path": "openai/whisper-tiny", "activation_dropout": 0.0, "activation_function": "gelu", - "architectures": [ - "WhisperForConditionalGeneration" - ], "attention_dropout": 0.0, "bos_token_id": 50257, "d_model": 384, @@ -12,7 +8,11 @@ "decoder_ffn_dim": 1536, "decoder_layerdrop": 0.0, "decoder_layers": 4, - "decoder_start_token_id": 50257, + "decoder_start_token_id": [ + 50258, + 50259, + 50359 + ], "dropout": 0.0, "encoder_attention_heads": 6, "encoder_ffn_dim": 1536, @@ -116,7 +116,6 @@ "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": 51865