MCPcopy Create free account

hub / github.com/argmaxinc/whisperkittools / functions

Functions200 in github.com/argmaxinc/whisperkittools

Methodbuild_cli
(self)
whisperkit/pipelines.py:466
Methodbuild_cli
(self)
whisperkit/pipelines.py:499
Methodbuild_cli
(self)
whisperkit/pipelines.py:538
Methodbuild_cli
(self)
whisperkit/pipelines.py:551
Methodbuild_cli
(self)
whisperkit/pipelines.py:630
Functioncli
f""" Generates Whisper models and publishes them to hf.co/{MODEL_REPO_ID}
scripts/generate_model.py:27
Methodclone_models
Download WhisperKit model files from Hugging Face Hub (only the files needed for `self.whisper_version`)
whisperkit/pipelines.py:155
Methodclone_models
Download whisper.cpp model files from Hugging Face Hub (only the ones needed for `self.whisper_version`)
whisperkit/pipelines.py:366
Methodclone_models
(self)
whisperkit/pipelines.py:470
Methodclone_models
(self)
whisperkit/pipelines.py:505
Methodclone_models
(self)
whisperkit/pipelines.py:557
Methodclone_models
(self)
whisperkit/pipelines.py:642
Methodclone_repo
(self)
whisperkit/pipelines.py:134
Methodclone_repo
(self)
whisperkit/pipelines.py:335
Methodclone_repo
(self)
whisperkit/pipelines.py:463
Methodclone_repo
(self)
whisperkit/pipelines.py:496
Methodclone_repo
(self)
whisperkit/pipelines.py:548
Methodclone_repo
(self)
whisperkit/pipelines.py:581
Methodcode_spec
(self)
whisperkit/test_utils.py:23
Methodcombine_cents
(m: Match)
whisperkit/evaluate/normalize_en.py:482
Functionconvert_tokenizer
Temporary function to reformat the original Whisper tokenizer
whisperkit/android/utils.py:210
Functionconvert_via_aihub
( model_name, torch_model, inputs, target_runtime="tflite", benchmark_
whisperkit/android/utils.py:96
Methoddivergence_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
Methoddivergence_fn
WhisperEncoder emits embeddings. The function used to quantify output change is negative PSNR (lower the better)
whisperkit/compress/palettize.py:109
Functionevaluate_folder
Evaluate a single audio folder with whisper_pipeline
whisperkit/evaluate/evaluate.py:280
Methodextract_cents
(m: Match)
whisperkit/evaluate/normalize_en.py:491
Functionfilter_audio_ext
(f)
whisperkit/evaluate/datasets.py:134
Methodforward
Shapes: Inputs: input_embeds: (batch_size, embed_dim, 1, encoder_seq_len) Outputs:
whisperkit/audio_encoder.py:47
Methodforward
Shapes: Inputs: melspectrogram_features: (batch_size, num_mel_bins, 1, encoder_seq_len * 2) Outpu
whisperkit/audio_encoder.py:95
Methodforward
(self, audio: tt.WhisperMelSpectrogramInputType )
whisperkit/audio_encoder.py:191
Methodforward
(self, hidden_states)
whisperkit/wavlm.py:45
Methodforward
(self, inputs)
whisperkit/wavlm.py:65
Methodforward
(self, hidden_states)
whisperkit/wavlm.py:81
Methodforward
(self, hidden_states)
whisperkit/wavlm.py:108
Methodforward
(self, hidden_states, attention_mask, position_bias)
whisperkit/wavlm.py:141
Methodforward
(self, hidden_states, attention_mask=None, position_bias=None,
whisperkit/wavlm.py:193
Methodforward
( self, hidden_states, attention_mask=None, )
whisperkit/wavlm.py:219
Methodforward
(self, hidden_states, positional_embeddings, attention_mask=None)
whisperkit/wavlm.py:244
Methodforward
(self, hidden_states, positional_embeddings, attention_mask=None)
whisperkit/wavlm.py:274
Methodforward
(self, waveforms)
whisperkit/wavlm.py:301
Methodforward
(self, hidden_states, positional_embeddings, attention_mask=None)
whisperkit/wavlm.py:342
Methodforward
(self, hidden_states)
whisperkit/wavlm.py:361
Methodforward
(self, input_values, attention_mask=None)
whisperkit/wavlm.py:433
Methodforward
(self, hidden_states, positional_embeddings, attention_mask=None)
whisperkit/wavlm.py:445
Methodforward
(self, input_values, attention_mask=None)
whisperkit/wavlm.py:477
Methodforward
(self, hidden_states, positional_embeddings, attention_mask=None)
whisperkit/wavlm.py:491
Methodforward
Shapes: Inputs: input_embeds: (batch_size, embed_dim, 1, q_seq_len=1) key_cache:
whisperkit/text_decoder.py:69
Methodforward
Shapes: Inputs: input_ids: (batch, ) cache_length: (batch, )
whisperkit/text_decoder.py:172
Methodforward
Look-up key-value cache prefill values for a given task spec. Args: task: 0->transcribe, 1->translate language:
whisperkit/text_decoder.py:485
Methodforward
(self, waveform: torch.Tensor, attention_mask=None)
whisperkit/diarization/sseriouss_speaker_segmenter.py:159
Methodforward
Apply STFT on input (audio) data
whisperkit/android/models.py:62
Methodforward
(self, audio: tt.WhisperMelSpectrogramInputType)
whisperkit/android/models.py:100
Methodforward
(self, logits)
whisperkit/android/models.py:118
Methodforward
Compute RMSE for audio waveform
whisperkit/android/models.py:135
Methodfrom_transformers
(cls, source: hf_wavlm.WavLMAttention)
whisperkit/wavlm.py:135
Methodinit_model_and_test_data
Initialize WhisperTextDecoder and a test mini-batch of random data
whisperkit/compress/palettize.py:27
Methodinit_model_and_test_data
Initialize WhisperAudioEncoder and obtain a test mini-batch of random data
whisperkit/compress/palettize.py:86
Functionlinear_to_conv2d_map_attention_wavlm
(state_dict, prefix, local_metadata, strict, missing_keys, unexpected
whisperkit/wavlm.py:528
Functionlinear_to_conv2d_map_ffn_wavlm
(state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys,
whisperkit/wavlm.py:540
Functionmap_join
(p)
whisperkit/evaluate/datasets.py:139
Methodmodel_spec
(self)
whisperkit/test_utils.py:26
Functionmove_weights_to_preprocessor
(state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, e
whisperkit/wavlm.py:499
Functionone_hot
(context, node)
whisperkit/diarization/diarization_utils.py:18
Methodoutput
(result: Union[str, int])
whisperkit/evaluate/normalize_en.py:240
Methodplace
(t)
tests/test_audio_encoder.py:88
Methodplot_specs
(self, f, ax)
whisperkit/compress/palettize.py:73
Methodplot_specs
(self, f, ax)
whisperkit/compress/palettize.py:115
Functionpreprocess_fn
(batch)
whisperkit/evaluate/datasets.py:81
Methodquant_variant
(self)
whisperkit/pipelines.py:343
Functionremove_symbols
Replace any other markers, symbols, punctuations with a space, keeping diacritics
whisperkit/evaluate/normalize_en.py:69
Methodsave_w
(module, input, output)
whisperkit/text_decoder.py:148
MethodsetUpClass
(cls)
tests/test_aihub.py:26
MethodsetUpClass
(cls)
tests/test_text_decoder.py:82
MethodsetUpClass
(cls)
tests/test_text_decoder.py:384
MethodsetUpClass
(cls)
tests/test_evaluate.py:43
MethodsetUpClass
(cls)
tests/test_audio_encoder.py:139
MethodsetUpClass
(cls)
tests/test_audio_encoder.py:206
MethodsetUpClass
(cls)
tests/test_word_timestamps.py:31
MethodsetUpClass
(cls)
tests/apple/speaker_segmenter/test_sseriouss.py:55
Functionset_metadata_for_whisper_decoder
Add metadata to Core ML models for traceability
whisperkit/test_utils.py:163
Functionsummarize_performance
Summarize the profile job results from AI Hub
whisperkit/android/utils.py:161
MethodtearDownClass
(cls)
tests/test_aihub.py:50
MethodtearDownClass
(cls)
tests/test_text_decoder.py:157
MethodtearDownClass
(cls)
tests/test_text_decoder.py:398
MethodtearDownClass
(cls)
tests/test_audio_encoder.py:170
MethodtearDownClass
(cls)
tests/test_audio_encoder.py:217
Methodtest_accuracy
(self)
tests/test_word_timestamps.py:57
Methodtest_evaluate
(self)
tests/test_evaluate.py:128
Methodtest_torch2aihub_performance_and_correctness
Test AI Hub compilation and inference job results against local PyTorch test results
tests/test_aihub.py:73
Methodtest_torch2torch_correctness
Test forward pass functionality and correctness of PyTorch models
tests/test_aihub.py:54
Methodtest_torch2torch_correctness
Coverage: - torch2torch parity transformers.models.whisper.modeling_whisper.WhisperDecoder and whisperkit.text_decoder.WhisperTextDeco
tests/test_text_decoder.py:176
Methodtest_torch2torch_correctness
Coverage: - torch2torch parity transformers.models.whisper.modeling_whisper.WhisperEncoder and argmax.ane.whisper.WhisperEncoder
tests/test_audio_encoder.py:120
Methodtest_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
Methodto_fraction
(s: str)
whisperkit/evaluate/normalize_en.py:234
Methodtranscribe
Transcribe an audio file using the WhisperKit CLI
whisperkit/pipelines.py:211
Methodtranscribe
Transcribe an audio file using the whisper.cpp CLI
whisperkit/pipelines.py:407
Methodtranscribe
Transcribe an audio file using MLX
whisperkit/pipelines.py:473
Methodtranscribe
Transcribe an audio file using PyTorch OpenAI Whisper
whisperkit/pipelines.py:561
Methodtranscribe
Transcribe an audio file using the Apple Speech Analyzer CLI
whisperkit/pipelines.py:647
Functionunfold
(context, node)
whisperkit/diarization/diarization_utils.py:25
← previous101–200 of 200, ranked by callers