Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/argmaxinc/whisperkittools
/ functions
Functions
200 in github.com/argmaxinc/whisperkittools
⨍
Functions
200
◇
Types & classes
56
Method
build_cli
(self)
whisperkit/pipelines.py:466
Method
build_cli
(self)
whisperkit/pipelines.py:499
Method
build_cli
(self)
whisperkit/pipelines.py:538
Method
build_cli
(self)
whisperkit/pipelines.py:551
Method
build_cli
(self)
whisperkit/pipelines.py:630
Function
cli
f""" Generates Whisper models and publishes them to hf.co/{MODEL_REPO_ID}
scripts/generate_model.py:27
Method
clone_models
Download WhisperKit model files from Hugging Face Hub (only the files needed for `self.whisper_version`)
whisperkit/pipelines.py:155
Method
clone_models
Download whisper.cpp model files from Hugging Face Hub (only the ones needed for `self.whisper_version`)
whisperkit/pipelines.py:366
Method
clone_models
(self)
whisperkit/pipelines.py:470
Method
clone_models
(self)
whisperkit/pipelines.py:505
Method
clone_models
(self)
whisperkit/pipelines.py:557
Method
clone_models
(self)
whisperkit/pipelines.py:642
Method
clone_repo
(self)
whisperkit/pipelines.py:134
Method
clone_repo
(self)
whisperkit/pipelines.py:335
Method
clone_repo
(self)
whisperkit/pipelines.py:463
Method
clone_repo
(self)
whisperkit/pipelines.py:496
Method
clone_repo
(self)
whisperkit/pipelines.py:548
Method
clone_repo
(self)
whisperkit/pipelines.py:581
Method
code_spec
(self)
whisperkit/test_utils.py:23
Method
combine_cents
(m: Match)
whisperkit/evaluate/normalize_en.py:482
Function
convert_tokenizer
Temporary function to reformat the original Whisper tokenizer
whisperkit/android/utils.py:210
Function
convert_via_aihub
( model_name, torch_model, inputs, target_runtime="tflite", benchmark_
whisperkit/android/utils.py:96
Method
divergence_fn
WhisperTextDecoder emits logits over a token vocabulary. The function used to quantify output change is KL divergence (lower the better)
whisperkit/compress/palettize.py:51
Method
divergence_fn
WhisperEncoder emits embeddings. The function used to quantify output change is negative PSNR (lower the better)
whisperkit/compress/palettize.py:109
Function
evaluate_folder
Evaluate a single audio folder with whisper_pipeline
whisperkit/evaluate/evaluate.py:280
Method
extract_cents
(m: Match)
whisperkit/evaluate/normalize_en.py:491
Function
filter_audio_ext
(f)
whisperkit/evaluate/datasets.py:134
Method
forward
Shapes: Inputs: input_embeds: (batch_size, embed_dim, 1, encoder_seq_len) Outputs:
whisperkit/audio_encoder.py:47
Method
forward
Shapes: Inputs: melspectrogram_features: (batch_size, num_mel_bins, 1, encoder_seq_len * 2) Outpu
whisperkit/audio_encoder.py:95
Method
forward
(self, audio: tt.WhisperMelSpectrogramInputType )
whisperkit/audio_encoder.py:191
Method
forward
(self, hidden_states)
whisperkit/wavlm.py:45
Method
forward
(self, inputs)
whisperkit/wavlm.py:65
Method
forward
(self, hidden_states)
whisperkit/wavlm.py:81
Method
forward
(self, hidden_states)
whisperkit/wavlm.py:108
Method
forward
(self, hidden_states, attention_mask, position_bias)
whisperkit/wavlm.py:141
Method
forward
(self, hidden_states, attention_mask=None, position_bias=None,
whisperkit/wavlm.py:193
Method
forward
( self, hidden_states, attention_mask=None, )
whisperkit/wavlm.py:219
Method
forward
(self, hidden_states, positional_embeddings, attention_mask=None)
whisperkit/wavlm.py:244
Method
forward
(self, hidden_states, positional_embeddings, attention_mask=None)
whisperkit/wavlm.py:274
Method
forward
(self, waveforms)
whisperkit/wavlm.py:301
Method
forward
(self, hidden_states, positional_embeddings, attention_mask=None)
whisperkit/wavlm.py:342
Method
forward
(self, hidden_states)
whisperkit/wavlm.py:361
Method
forward
(self, input_values, attention_mask=None)
whisperkit/wavlm.py:433
Method
forward
(self, hidden_states, positional_embeddings, attention_mask=None)
whisperkit/wavlm.py:445
Method
forward
(self, input_values, attention_mask=None)
whisperkit/wavlm.py:477
Method
forward
(self, hidden_states, positional_embeddings, attention_mask=None)
whisperkit/wavlm.py:491
Method
forward
Shapes: Inputs: input_embeds: (batch_size, embed_dim, 1, q_seq_len=1) key_cache:
whisperkit/text_decoder.py:69
Method
forward
Shapes: Inputs: input_ids: (batch, ) cache_length: (batch, )
whisperkit/text_decoder.py:172
Method
forward
Look-up key-value cache prefill values for a given task spec. Args: task: 0->transcribe, 1->translate language:
whisperkit/text_decoder.py:485
Method
forward
(self, waveform: torch.Tensor, attention_mask=None)
whisperkit/diarization/sseriouss_speaker_segmenter.py:159
Method
forward
Apply STFT on input (audio) data
whisperkit/android/models.py:62
Method
forward
(self, audio: tt.WhisperMelSpectrogramInputType)
whisperkit/android/models.py:100
Method
forward
(self, logits)
whisperkit/android/models.py:118
Method
forward
Compute RMSE for audio waveform
whisperkit/android/models.py:135
Method
from_transformers
(cls, source: hf_wavlm.WavLMAttention)
whisperkit/wavlm.py:135
Method
init_model_and_test_data
Initialize WhisperTextDecoder and a test mini-batch of random data
whisperkit/compress/palettize.py:27
Method
init_model_and_test_data
Initialize WhisperAudioEncoder and obtain a test mini-batch of random data
whisperkit/compress/palettize.py:86
Function
linear_to_conv2d_map_attention_wavlm
(state_dict, prefix, local_metadata, strict, missing_keys, unexpected
whisperkit/wavlm.py:528
Function
linear_to_conv2d_map_ffn_wavlm
(state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys,
whisperkit/wavlm.py:540
Function
map_join
(p)
whisperkit/evaluate/datasets.py:139
Method
model_spec
(self)
whisperkit/test_utils.py:26
Function
move_weights_to_preprocessor
(state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, e
whisperkit/wavlm.py:499
Function
one_hot
(context, node)
whisperkit/diarization/diarization_utils.py:18
Method
output
(result: Union[str, int])
whisperkit/evaluate/normalize_en.py:240
Method
place
(t)
tests/test_audio_encoder.py:88
Method
plot_specs
(self, f, ax)
whisperkit/compress/palettize.py:73
Method
plot_specs
(self, f, ax)
whisperkit/compress/palettize.py:115
Function
preprocess_fn
(batch)
whisperkit/evaluate/datasets.py:81
Method
quant_variant
(self)
whisperkit/pipelines.py:343
Function
remove_symbols
Replace any other markers, symbols, punctuations with a space, keeping diacritics
whisperkit/evaluate/normalize_en.py:69
Method
save_w
(module, input, output)
whisperkit/text_decoder.py:148
Method
setUpClass
(cls)
tests/test_aihub.py:26
Method
setUpClass
(cls)
tests/test_text_decoder.py:82
Method
setUpClass
(cls)
tests/test_text_decoder.py:384
Method
setUpClass
(cls)
tests/test_evaluate.py:43
Method
setUpClass
(cls)
tests/test_audio_encoder.py:139
Method
setUpClass
(cls)
tests/test_audio_encoder.py:206
Method
setUpClass
(cls)
tests/test_word_timestamps.py:31
Method
setUpClass
(cls)
tests/apple/speaker_segmenter/test_sseriouss.py:55
Function
set_metadata_for_whisper_decoder
Add metadata to Core ML models for traceability
whisperkit/test_utils.py:163
Function
summarize_performance
Summarize the profile job results from AI Hub
whisperkit/android/utils.py:161
Method
tearDownClass
(cls)
tests/test_aihub.py:50
Method
tearDownClass
(cls)
tests/test_text_decoder.py:157
Method
tearDownClass
(cls)
tests/test_text_decoder.py:398
Method
tearDownClass
(cls)
tests/test_audio_encoder.py:170
Method
tearDownClass
(cls)
tests/test_audio_encoder.py:217
Method
test_accuracy
(self)
tests/test_word_timestamps.py:57
Method
test_evaluate
(self)
tests/test_evaluate.py:128
Method
test_torch2aihub_performance_and_correctness
Test AI Hub compilation and inference job results against local PyTorch test results
tests/test_aihub.py:73
Method
test_torch2torch_correctness
Test forward pass functionality and correctness of PyTorch models
tests/test_aihub.py:54
Method
test_torch2torch_correctness
Coverage: - torch2torch parity transformers.models.whisper.modeling_whisper.WhisperDecoder and whisperkit.text_decoder.WhisperTextDeco
tests/test_text_decoder.py:176
Method
test_torch2torch_correctness
Coverage: - torch2torch parity transformers.models.whisper.modeling_whisper.WhisperEncoder and argmax.ane.whisper.WhisperEncoder
tests/test_audio_encoder.py:120
Method
test_torch_context_prefill
Coverage: - Build a WhisperDecoderContextPrefill module - Use a test WhisperDecoder instance to compute all valid context pre
tests/test_text_decoder.py:201
Method
to_fraction
(s: str)
whisperkit/evaluate/normalize_en.py:234
Method
transcribe
Transcribe an audio file using the WhisperKit CLI
whisperkit/pipelines.py:211
Method
transcribe
Transcribe an audio file using the whisper.cpp CLI
whisperkit/pipelines.py:407
Method
transcribe
Transcribe an audio file using MLX
whisperkit/pipelines.py:473
Method
transcribe
Transcribe an audio file using PyTorch OpenAI Whisper
whisperkit/pipelines.py:561
Method
transcribe
Transcribe an audio file using the Apple Speech Analyzer CLI
whisperkit/pipelines.py:647
Function
unfold
(context, node)
whisperkit/diarization/diarization_utils.py:25
← previous
101–200 of 200, ranked by callers