Migrate to zeromodels (rename kf_*.json -> zm_*.json, fix refs in config + README, ensure tag + badge)

README.md

@@ -2,10 +2,10 @@
 pipeline_tag: automatic-speech-recognition
 license: apache-2.0
 base_model: openai/whisper-large-v2
-library_name: kerasformers
+library_name: zeromodels
 tags:
 - keras
-- kerasformers
+- zeromodels
 - whisper
 - automatic-speech-recognition
 - audio
@@ -16,13 +16,13 @@ tags:
 - tf
 ---
 
-## ***See [our collection](https://huggingface.co/collections/kerasformers/whisper-6a6ac856710a03d7d2e207b6) for all versions of Whisper.***
+## ***See [our collection](https://huggingface.co/collections/zeromodels/whisper-6a6ac856710a03d7d2e207b6) for all versions of Whisper.***
 
 # Run Whisper with Keras 3: JAX, PyTorch, or TensorFlow
 
-[![GitHub](https://img.shields.io/badge/GitHub-KerasFormers-black?logo=github)](https://github.com/IMvision12/KerasFormers) [![Docs](https://img.shields.io/badge/Docs-Whisper-blue)](https://imvision12.github.io/KerasFormers/whisper/) [![Collection](https://img.shields.io/badge/HF-Whisper%20collection-yellow)](https://huggingface.co/collections/kerasformers/whisper-6a6ac856710a03d7d2e207b6)
+[![GitHub](https://img.shields.io/badge/GitHub-ZeroModels-black?logo=github)](https://github.com/IMvision12/ZeroModels) [![Docs](https://img.shields.io/badge/Docs-Whisper-blue)](https://imvision12.github.io/ZeroModels/whisper/) [![Collection](https://img.shields.io/badge/HF-Whisper%20collection-yellow)](https://huggingface.co/collections/zeromodels/whisper-6a6ac856710a03d7d2e207b6)
 
-# kerasformers/whisper_large_v2
+# zeromodels/whisper_large_v2
 
 Paper: [Robust Speech Recognition via Large-Scale Weak Supervision (arXiv:2212.04356)](https://arxiv.org/abs/2212.04356) · [HF Papers](https://huggingface.co/papers/2212.04356)
 
@@ -30,7 +30,7 @@ Whisper is a **multilingual** encoder-decoder ASR model trained on large-scale w
 
 For more details on the model, please go to the upstream [model card](https://huggingface.co/openai/whisper-large-v2).
 
-Pure-**Keras 3** conversion of [`openai/whisper-large-v2`](https://huggingface.co/openai/whisper-large-v2) for [kerasformers](https://github.com/IMvision12/KerasFormers). One implementation runs unmodified on **TensorFlow / Torch / JAX**.
+Pure-**Keras 3** conversion of [`openai/whisper-large-v2`](https://huggingface.co/openai/whisper-large-v2) for [zeromodels](https://github.com/IMvision12/ZeroModels). One implementation runs unmodified on **TensorFlow / Torch / JAX**.
 
 This is an **ASR** checkpoint (`WhisperConditionalGenerate`, 1.55B).
 
@@ -41,13 +41,13 @@ import os
 os.environ["KERAS_BACKEND"] = "torch"  # or "jax" / "tensorflow"
 
 import soundfile as sf
-from kerasformers.models.whisper import (
+from zeromodels.models.whisper import (
     WhisperProcessor,
     WhisperConditionalGenerate,
 )
 
-model = WhisperConditionalGenerate.from_weights("kerasformers/whisper_large_v2")
-processor = WhisperProcessor.from_weights("kerasformers/whisper_large_v2")
+model = WhisperConditionalGenerate.from_weights("zeromodels/whisper_large_v2")
+processor = WhisperProcessor.from_weights("zeromodels/whisper_large_v2")
 
 audio, sr = sf.read("your_audio.wav", dtype="float32")  # 16 kHz mono
 # task="transcribe" keeps the source language; "translate" -> English.
@@ -55,25 +55,25 @@ text = model.generate(audio, processor, language="en", task="transcribe")
 print(repr(text[0]))
 ```
 
-Load any Whisper variant the same way with `from_weights("kerasformers/<variant>")`:
+Load any Whisper variant the same way with `from_weights("zeromodels/<variant>")`:
 
 | Variant | Hub | Notes |
 |---|---|---|
-| `whisper_tiny` | [`kerasformers/whisper_tiny`](https://huggingface.co/kerasformers/whisper_tiny) | 39M |
-| `whisper_base` | [`kerasformers/whisper_base`](https://huggingface.co/kerasformers/whisper_base) | 74M |
-| `whisper_small` | [`kerasformers/whisper_small`](https://huggingface.co/kerasformers/whisper_small) | 244M |
-| `whisper_medium` | [`kerasformers/whisper_medium`](https://huggingface.co/kerasformers/whisper_medium) | 769M |
-| `whisper_large` | [`kerasformers/whisper_large`](https://huggingface.co/kerasformers/whisper_large) | 1.55B |
-| `whisper_large_v2` | [`kerasformers/whisper_large_v2`](https://huggingface.co/kerasformers/whisper_large_v2) | 1.55B |
-| `whisper_large_v3` | [`kerasformers/whisper_large_v3`](https://huggingface.co/kerasformers/whisper_large_v3) | 128 mel bins |
-| `whisper_large_v3_turbo` | [`kerasformers/whisper_large_v3_turbo`](https://huggingface.co/kerasformers/whisper_large_v3_turbo) | 4 decoder layers |
+| `whisper_tiny` | [`zeromodels/whisper_tiny`](https://huggingface.co/zeromodels/whisper_tiny) | 39M |
+| `whisper_base` | [`zeromodels/whisper_base`](https://huggingface.co/zeromodels/whisper_base) | 74M |
+| `whisper_small` | [`zeromodels/whisper_small`](https://huggingface.co/zeromodels/whisper_small) | 244M |
+| `whisper_medium` | [`zeromodels/whisper_medium`](https://huggingface.co/zeromodels/whisper_medium) | 769M |
+| `whisper_large` | [`zeromodels/whisper_large`](https://huggingface.co/zeromodels/whisper_large) | 1.55B |
+| `whisper_large_v2` | [`zeromodels/whisper_large_v2`](https://huggingface.co/zeromodels/whisper_large_v2) | 1.55B |
+| `whisper_large_v3` | [`zeromodels/whisper_large_v3`](https://huggingface.co/zeromodels/whisper_large_v3) | 128 mel bins |
+| `whisper_large_v3_turbo` | [`zeromodels/whisper_large_v3_turbo`](https://huggingface.co/zeromodels/whisper_large_v3_turbo) | 4 decoder layers |
 
 ## Tips
 
-- Set `KERAS_BACKEND` **before** importing Keras / kerasformers.
+- Set `KERAS_BACKEND` **before** importing Keras / zeromodels.
 - Prefer `WhisperProcessor.from_weights(...)` so mel bins match the variant (v3 uses 128).
 - Clips are padded to a 30 s window; chunk longer audio yourself.
-- See [Whisper docs](https://imvision12.github.io/KerasFormers/whisper/) and [Loading Weights](https://imvision12.github.io/KerasFormers/loading_weights/).
+- See [Whisper docs](https://imvision12.github.io/ZeroModels/whisper/) and [Loading Weights](https://imvision12.github.io/ZeroModels/loading_weights/).
 - Community / upstream safetensors still work via the `hf:` prefix, e.g. `WhisperConditionalGenerate.from_weights("hf:openai/whisper-large-v2")`.
 
 ## Special Thanks

kf_config.json → zm_config.json

@@ -1,344 +1,344 @@
-{
-  "library_name": "kerasformers",
-  "kerasformers_version": "1.2.1",
-  "model_module": "kerasformers.models.whisper",
-  "model_class": "WhisperModel",
-  "variant": "whisper_large_v2",
-  "weights": "model.weights.json",
-  "schema_version": 2,
-  "weight_dtype": "float32",
-  "model_type": "whisper",
-  "audio_config": {
-    "num_layers": 32,
-    "attention_heads": 20,
-    "ffn_dim": 5120,
-    "num_mel_bins": 80,
-    "max_source_positions": 1500
-  },
-  "text_config": {
-    "num_layers": 32,
-    "attention_heads": 20,
-    "ffn_dim": 5120,
-    "vocab_size": 51865,
-    "max_target_positions": 448
-  },
-  "hidden_dim": 1280,
-  "activation_function": "gelu",
-  "layer_norm_eps": 1e-05,
-  "scale_embedding": false,
-  "generate_args": {
-    "alignment_heads": [
-      [
-        10,
-        12
-      ],
-      [
-        13,
-        17
-      ],
-      [
-        16,
-        11
-      ],
-      [
-        16,
-        12
-      ],
-      [
-        16,
-        13
-      ],
-      [
-        17,
-        15
-      ],
-      [
-        17,
-        16
-      ],
-      [
-        18,
-        4
-      ],
-      [
-        18,
-        11
-      ],
-      [
-        18,
-        19
-      ],
-      [
-        19,
-        11
-      ],
-      [
-        21,
-        2
-      ],
-      [
-        21,
-        3
-      ],
-      [
-        22,
-        3
-      ],
-      [
-        22,
-        9
-      ],
-      [
-        22,
-        12
-      ],
-      [
-        23,
-        5
-      ],
-      [
-        23,
-        7
-      ],
-      [
-        23,
-        13
-      ],
-      [
-        25,
-        5
-      ],
-      [
-        26,
-        1
-      ],
-      [
-        26,
-        12
-      ],
-      [
-        27,
-        15
-      ]
-    ],
-    "begin_suppress_tokens": [
-      220,
-      50257
-    ],
-    "bos_token_id": 50257,
-    "decoder_start_token_id": 50258,
-    "eos_token_id": 50257,
-    "forced_decoder_ids": [
-      [
-        1,
-        null
-      ],
-      [
-        2,
-        50359
-      ]
-    ],
-    "is_multilingual": true,
-    "lang_to_id": {
-      "<|af|>": 50327,
-      "<|am|>": 50334,
-      "<|ar|>": 50272,
-      "<|as|>": 50350,
-      "<|az|>": 50304,
-      "<|ba|>": 50355,
-      "<|be|>": 50330,
-      "<|bg|>": 50292,
-      "<|bn|>": 50302,
-      "<|bo|>": 50347,
-      "<|br|>": 50309,
-      "<|bs|>": 50315,
-      "<|ca|>": 50270,
-      "<|cs|>": 50283,
-      "<|cy|>": 50297,
-      "<|da|>": 50285,
-      "<|de|>": 50261,
-      "<|el|>": 50281,
-      "<|en|>": 50259,
-      "<|es|>": 50262,
-      "<|et|>": 50307,
-      "<|eu|>": 50310,
-      "<|fa|>": 50300,
-      "<|fi|>": 50277,
-      "<|fo|>": 50338,
-      "<|fr|>": 50265,
-      "<|gl|>": 50319,
-      "<|gu|>": 50333,
-      "<|haw|>": 50352,
-      "<|ha|>": 50354,
-      "<|he|>": 50279,
-      "<|hi|>": 50276,
-      "<|hr|>": 50291,
-      "<|ht|>": 50339,
-      "<|hu|>": 50286,
-      "<|hy|>": 50312,
-      "<|id|>": 50275,
-      "<|is|>": 50311,
-      "<|it|>": 50274,
-      "<|ja|>": 50266,
-      "<|jw|>": 50356,
-      "<|ka|>": 50329,
-      "<|kk|>": 50316,
-      "<|km|>": 50323,
-      "<|kn|>": 50306,
-      "<|ko|>": 50264,
-      "<|la|>": 50294,
-      "<|lb|>": 50345,
-      "<|ln|>": 50353,
-      "<|lo|>": 50336,
-      "<|lt|>": 50293,
-      "<|lv|>": 50301,
-      "<|mg|>": 50349,
-      "<|mi|>": 50295,
-      "<|mk|>": 50308,
-      "<|ml|>": 50296,
-      "<|mn|>": 50314,
-      "<|mr|>": 50320,
-      "<|ms|>": 50282,
-      "<|mt|>": 50343,
-      "<|my|>": 50346,
-      "<|ne|>": 50313,
-      "<|nl|>": 50271,
-      "<|nn|>": 50342,
-      "<|no|>": 50288,
-      "<|oc|>": 50328,
-      "<|pa|>": 50321,
-      "<|pl|>": 50269,
-      "<|ps|>": 50340,
-      "<|pt|>": 50267,
-      "<|ro|>": 50284,
-      "<|ru|>": 50263,
-      "<|sa|>": 50344,
-      "<|sd|>": 50332,
-      "<|si|>": 50322,
-      "<|sk|>": 50298,
-      "<|sl|>": 50305,
-      "<|sn|>": 50324,
-      "<|so|>": 50326,
-      "<|sq|>": 50317,
-      "<|sr|>": 50303,
-      "<|su|>": 50357,
-      "<|sv|>": 50273,
-      "<|sw|>": 50318,
-      "<|ta|>": 50287,
-      "<|te|>": 50299,
-      "<|tg|>": 50331,
-      "<|th|>": 50289,
-      "<|tk|>": 50341,
-      "<|tl|>": 50348,
-      "<|tr|>": 50268,
-      "<|tt|>": 50351,
-      "<|uk|>": 50280,
-      "<|ur|>": 50290,
-      "<|uz|>": 50337,
-      "<|vi|>": 50278,
-      "<|yi|>": 50335,
-      "<|yo|>": 50325,
-      "<|zh|>": 50260
-    },
-    "max_initial_timestamp_index": 50,
-    "max_length": 448,
-    "no_timestamps_token_id": 50363,
-    "pad_token_id": 50257,
-    "prev_sot_token_id": 50361,
-    "return_timestamps": false,
-    "suppress_tokens": [
-      1,
-      2,
-      7,
-      8,
-      9,
-      10,
-      14,
-      25,
-      26,
-      27,
-      28,
-      29,
-      31,
-      58,
-      59,
-      60,
-      61,
-      62,
-      63,
-      90,
-      91,
-      92,
-      93,
-      359,
-      503,
-      522,
-      542,
-      873,
-      893,
-      902,
-      918,
-      922,
-      931,
-      1350,
-      1853,
-      1982,
-      2460,
-      2627,
-      3246,
-      3253,
-      3268,
-      3536,
-      3846,
-      3961,
-      4183,
-      4667,
-      6585,
-      6647,
-      7273,
-      9061,
-      9383,
-      10428,
-      10929,
-      11938,
-      12033,
-      12331,
-      12562,
-      13793,
-      14157,
-      14635,
-      15265,
-      15618,
-      16553,
-      16604,
-      18362,
-      18956,
-      20075,
-      21675,
-      22520,
-      26130,
-      26161,
-      26435,
-      28279,
-      29464,
-      31650,
-      32302,
-      32470,
-      36865,
-      42863,
-      47425,
-      49870,
-      50254,
-      50258,
-      50358,
-      50359,
-      50360,
-      50361,
-      50362
-    ],
-    "task_to_id": {
-      "transcribe": 50359,
-      "translate": 50358
-    }
-  }
+{
+  "library_name": "zeromodels",
+  "zeromodels_version": "1.2.1",
+  "model_module": "zeromodels.models.whisper",
+  "model_class": "WhisperModel",
+  "variant": "whisper_large_v2",
+  "weights": "model.weights.json",
+  "schema_version": 2,
+  "weight_dtype": "float32",
+  "model_type": "whisper",
+  "audio_config": {
+    "num_layers": 32,
+    "attention_heads": 20,
+    "ffn_dim": 5120,
+    "num_mel_bins": 80,
+    "max_source_positions": 1500
+  },
+  "text_config": {
+    "num_layers": 32,
+    "attention_heads": 20,
+    "ffn_dim": 5120,
+    "vocab_size": 51865,
+    "max_target_positions": 448
+  },
+  "hidden_dim": 1280,
+  "activation_function": "gelu",
+  "layer_norm_eps": 1e-05,
+  "scale_embedding": false,
+  "generate_args": {
+    "alignment_heads": [
+      [
+        10,
+        12
+      ],
+      [
+        13,
+        17
+      ],
+      [
+        16,
+        11
+      ],
+      [
+        16,
+        12
+      ],
+      [
+        16,
+        13
+      ],
+      [
+        17,
+        15
+      ],
+      [
+        17,
+        16
+      ],
+      [
+        18,
+        4
+      ],
+      [
+        18,
+        11
+      ],
+      [
+        18,
+        19
+      ],
+      [
+        19,
+        11
+      ],
+      [
+        21,
+        2
+      ],
+      [
+        21,
+        3
+      ],
+      [
+        22,
+        3
+      ],
+      [
+        22,
+        9
+      ],
+      [
+        22,
+        12
+      ],
+      [
+        23,
+        5
+      ],
+      [
+        23,
+        7
+      ],
+      [
+        23,
+        13
+      ],
+      [
+        25,
+        5
+      ],
+      [
+        26,
+        1
+      ],
+      [
+        26,
+        12
+      ],
+      [
+        27,
+        15
+      ]
+    ],
+    "begin_suppress_tokens": [
+      220,
+      50257
+    ],
+    "bos_token_id": 50257,
+    "decoder_start_token_id": 50258,
+    "eos_token_id": 50257,
+    "forced_decoder_ids": [
+      [
+        1,
+        null
+      ],
+      [
+        2,
+        50359
+      ]
+    ],
+    "is_multilingual": true,
+    "lang_to_id": {
+      "<|af|>": 50327,
+      "<|am|>": 50334,
+      "<|ar|>": 50272,
+      "<|as|>": 50350,
+      "<|az|>": 50304,
+      "<|ba|>": 50355,
+      "<|be|>": 50330,
+      "<|bg|>": 50292,
+      "<|bn|>": 50302,
+      "<|bo|>": 50347,
+      "<|br|>": 50309,
+      "<|bs|>": 50315,
+      "<|ca|>": 50270,
+      "<|cs|>": 50283,
+      "<|cy|>": 50297,
+      "<|da|>": 50285,
+      "<|de|>": 50261,
+      "<|el|>": 50281,
+      "<|en|>": 50259,
+      "<|es|>": 50262,
+      "<|et|>": 50307,
+      "<|eu|>": 50310,
+      "<|fa|>": 50300,
+      "<|fi|>": 50277,
+      "<|fo|>": 50338,
+      "<|fr|>": 50265,
+      "<|gl|>": 50319,
+      "<|gu|>": 50333,
+      "<|haw|>": 50352,
+      "<|ha|>": 50354,
+      "<|he|>": 50279,
+      "<|hi|>": 50276,
+      "<|hr|>": 50291,
+      "<|ht|>": 50339,
+      "<|hu|>": 50286,
+      "<|hy|>": 50312,
+      "<|id|>": 50275,
+      "<|is|>": 50311,
+      "<|it|>": 50274,
+      "<|ja|>": 50266,
+      "<|jw|>": 50356,
+      "<|ka|>": 50329,
+      "<|kk|>": 50316,
+      "<|km|>": 50323,
+      "<|kn|>": 50306,
+      "<|ko|>": 50264,
+      "<|la|>": 50294,
+      "<|lb|>": 50345,
+      "<|ln|>": 50353,
+      "<|lo|>": 50336,
+      "<|lt|>": 50293,
+      "<|lv|>": 50301,
+      "<|mg|>": 50349,
+      "<|mi|>": 50295,
+      "<|mk|>": 50308,
+      "<|ml|>": 50296,
+      "<|mn|>": 50314,
+      "<|mr|>": 50320,
+      "<|ms|>": 50282,
+      "<|mt|>": 50343,
+      "<|my|>": 50346,
+      "<|ne|>": 50313,
+      "<|nl|>": 50271,
+      "<|nn|>": 50342,
+      "<|no|>": 50288,
+      "<|oc|>": 50328,
+      "<|pa|>": 50321,
+      "<|pl|>": 50269,
+      "<|ps|>": 50340,
+      "<|pt|>": 50267,
+      "<|ro|>": 50284,
+      "<|ru|>": 50263,
+      "<|sa|>": 50344,
+      "<|sd|>": 50332,
+      "<|si|>": 50322,
+      "<|sk|>": 50298,
+      "<|sl|>": 50305,
+      "<|sn|>": 50324,
+      "<|so|>": 50326,
+      "<|sq|>": 50317,
+      "<|sr|>": 50303,
+      "<|su|>": 50357,
+      "<|sv|>": 50273,
+      "<|sw|>": 50318,
+      "<|ta|>": 50287,
+      "<|te|>": 50299,
+      "<|tg|>": 50331,
+      "<|th|>": 50289,
+      "<|tk|>": 50341,
+      "<|tl|>": 50348,
+      "<|tr|>": 50268,
+      "<|tt|>": 50351,
+      "<|uk|>": 50280,
+      "<|ur|>": 50290,
+      "<|uz|>": 50337,
+      "<|vi|>": 50278,
+      "<|yi|>": 50335,
+      "<|yo|>": 50325,
+      "<|zh|>": 50260
+    },
+    "max_initial_timestamp_index": 50,
+    "max_length": 448,
+    "no_timestamps_token_id": 50363,
+    "pad_token_id": 50257,
+    "prev_sot_token_id": 50361,
+    "return_timestamps": false,
+    "suppress_tokens": [
+      1,
+      2,
+      7,
+      8,
+      9,
+      10,
+      14,
+      25,
+      26,
+      27,
+      28,
+      29,
+      31,
+      58,
+      59,
+      60,
+      61,
+      62,
+      63,
+      90,
+      91,
+      92,
+      93,
+      359,
+      503,
+      522,
+      542,
+      873,
+      893,
+      902,
+      918,
+      922,
+      931,
+      1350,
+      1853,
+      1982,
+      2460,
+      2627,
+      3246,
+      3253,
+      3268,
+      3536,
+      3846,
+      3961,
+      4183,
+      4667,
+      6585,
+      6647,
+      7273,
+      9061,
+      9383,
+      10428,
+      10929,
+      11938,
+      12033,
+      12331,
+      12562,
+      13793,
+      14157,
+      14635,
+      15265,
+      15618,
+      16553,
+      16604,
+      18362,
+      18956,
+      20075,
+      21675,
+      22520,
+      26130,
+      26161,
+      26435,
+      28279,
+      29464,
+      31650,
+      32302,
+      32470,
+      36865,
+      42863,
+      47425,
+      49870,
+      50254,
+      50258,
+      50358,
+      50359,
+      50360,
+      50361,
+      50362
+    ],
+    "task_to_id": {
+      "transcribe": 50359,
+      "translate": 50358
+    }
+  }
 }

kf_preprocessor.json → zm_preprocessor.json

@@ -1,12 +1,12 @@
-{
-  "library_name": "kerasformers",
-  "kerasformers_version": "1.1.3",
-  "preprocessor_module": "kerasformers.models.whisper",
-  "preprocessor_class": "WhisperFeatureExtractor",
-  "variant": "whisper_large_v2",
-  "sampling_rate": 16000,
-  "n_fft": 400,
-  "hop_length": 160,
-  "n_mels": 80,
-  "chunk_length": 30
+{
+  "library_name": "zeromodels",
+  "zeromodels_version": "1.1.3",
+  "preprocessor_module": "zeromodels.models.whisper",
+  "preprocessor_class": "WhisperFeatureExtractor",
+  "variant": "whisper_large_v2",
+  "sampling_rate": 16000,
+  "n_fft": 400,
+  "hop_length": 160,
+  "n_mels": 80,
+  "chunk_length": 30
 }