Upload WhisperForConditionalGeneration

2022-09-27 07:02:42 +00:00 · 2022-09-27 07:02:42 +00:00 · d902ab9a80
parent 25da2ff16c
commit d902ab9a80
2 changed files with 5 additions and 8 deletions
--- 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,
--- a/pytorch_model.bin
+++ b/pytorch_model.bin