From c5758d766c1323aa26eed7273109a97265090075 Mon Sep 17 00:00:00 2001 From: Matt Watson Date: Wed, 29 Nov 2023 18:38:03 -0800 Subject: [PATCH] Fix typo in whisper presets files --- keras_nlp/models/whisper/whisper_presets.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/keras_nlp/models/whisper/whisper_presets.py b/keras_nlp/models/whisper/whisper_presets.py index e8c0d075a4..b740ed833d 100644 --- a/keras_nlp/models/whisper/whisper_presets.py +++ b/keras_nlp/models/whisper/whisper_presets.py @@ -234,11 +234,11 @@ "special_tokens": ENGLISH_SPECIAL_TOKENS, "language_tokens": None, }, - "weights_url": "https://storage.googleapis.com/keras-nlp/models/whisper_base_en/v1/model.h5", + "weights_url": "https://storage.googleapis.com/keras-nlp/models/whisper_small_en/v1/model.h5", "weights_hash": "b75a89225e20019d85ff5f1c362f8a49", - "vocabulary_url": "https://storage.googleapis.com/keras-nlp/models/whisper_base_en/v1/vocab.json", + "vocabulary_url": "https://storage.googleapis.com/keras-nlp/models/whisper_small_en/v1/vocab.json", "vocabulary_hash": "22377f841debacb023848b3468ea3281", - "merges_url": "https://storage.googleapis.com/keras-nlp/models/whisper_base_en/v1/merges.txt", + "merges_url": "https://storage.googleapis.com/keras-nlp/models/whisper_small_en/v1/merges.txt", "merges_hash": "093ecf3f30371012f2e96fcfb10ea6ab", }, "whisper_medium_en": { @@ -370,11 +370,11 @@ "special_tokens": MULTILINGUAL_SPECIAL_TOKENS, "language_tokens": LANGUAGE_TOKENS, }, - "weights_url": "https://storage.googleapis.com/keras-nlp/models/whisper_base_multi/v1/model.h5", + "weights_url": "https://storage.googleapis.com/keras-nlp/models/whisper_small_multi/v1/model.h5", "weights_hash": "c90c6a895e522056b77b924b6e907ed8", - "vocabulary_url": "https://storage.googleapis.com/keras-nlp/models/whisper_base_multi/v1/vocab.json", + "vocabulary_url": "https://storage.googleapis.com/keras-nlp/models/whisper_small_multi/v1/vocab.json", "vocabulary_hash": "1b87ed3e3ecd9ccfdca74e64cbe81d68", - "merges_url": "https://storage.googleapis.com/keras-nlp/models/whisper_base_multi/v1/merges.txt", + "merges_url": "https://storage.googleapis.com/keras-nlp/models/whisper_small_multi/v1/merges.txt", "merges_hash": "c7f01d4100f6211417988889bf35ccd8", }, "whisper_medium_multi": {