From f1baaf0c070fd03fc67d773bebeff75023422b6d Mon Sep 17 00:00:00 2001 From: Yoach Lacombe Date: Tue, 1 Oct 2024 07:44:09 +0000 Subject: [PATCH] Upload WhisperForConditionalGeneration --- config.json | 49 +++++++++++++++++++++++++++++++++++++++++++++++ model.safetensors | 3 +++ 2 files changed, 52 insertions(+) create mode 100644 config.json create mode 100644 model.safetensors diff --git a/config.json b/config.json new file mode 100644 index 0000000..5904e05 --- /dev/null +++ b/config.json @@ -0,0 +1,49 @@ +{ + "_name_or_path": "/raid/yoach/tmp_whisper_turbo", + "activation_dropout": 0.0, + "activation_function": "gelu", + "apply_spec_augment": false, + "architectures": [ + "WhisperForConditionalGeneration" + ], + "attention_dropout": 0.0, + "begin_suppress_tokens": [ + 220, + 50256 + ], + "bos_token_id": 50257, + "classifier_proj_size": 256, + "d_model": 1280, + "decoder_attention_heads": 20, + "decoder_ffn_dim": 5120, + "decoder_layerdrop": 0.0, + "decoder_layers": 4, + "decoder_start_token_id": 50258, + "dropout": 0.0, + "encoder_attention_heads": 20, + "encoder_ffn_dim": 5120, + "encoder_layerdrop": 0.0, + "encoder_layers": 32, + "eos_token_id": 50257, + "init_std": 0.02, + "is_encoder_decoder": true, + "mask_feature_length": 10, + "mask_feature_min_masks": 0, + "mask_feature_prob": 0.0, + "mask_time_length": 10, + "mask_time_min_masks": 2, + "mask_time_prob": 0.05, + "max_source_positions": 1500, + "max_target_positions": 448, + "median_filter_width": 7, + "model_type": "whisper", + "num_hidden_layers": 32, + "num_mel_bins": 128, + "pad_token_id": 50257, + "scale_embedding": false, + "torch_dtype": "float32", + "transformers_version": "4.46.0.dev0", + "use_cache": true, + "use_weighted_layer_sum": false, + "vocab_size": 51866 +} diff --git a/model.safetensors b/model.safetensors new file mode 100644 index 0000000..f94834b --- /dev/null +++ b/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d4f0531bf268b0db8416c81887739b96b290fc2994caa093c18666e5b5ad0e83 +size 3235581408