From 8662dcd0ffbcd8f5128b2762ccd1823e52018e4d Mon Sep 17 00:00:00 2001 From: Arthur Zucker Date: Mon, 17 Oct 2022 11:06:59 +0000 Subject: [PATCH] Update `max_length` param --- config.json | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/config.json b/config.json index 3bdd412..ddd4331 100644 --- a/config.json +++ b/config.json @@ -1,6 +1,10 @@ { + "_name_or_path": "openai/whisper-tiny", "activation_dropout": 0.0, "activation_function": "gelu", + "architectures": [ + "WhisperForConditionalGeneration" + ], "attention_dropout": 0.0, "begin_suppress_tokens": [ 220, @@ -35,6 +39,7 @@ ], "init_std": 0.02, "is_encoder_decoder": true, + "max_length": 448, "max_source_positions": 1500, "max_target_positions": 448, "model_type": "whisper", @@ -132,7 +137,8 @@ 50361, 50362 ], - "transformers_version": "4.23.0.dev0", + "torch_dtype": "float32", + "transformers_version": "4.24.0.dev0", "use_cache": true, "vocab_size": 51865 }