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 }