MCPcopy Create free account

hub / github.com/FireRedTeam/FireRedASR / functions

Functions138 in github.com/FireRedTeam/FireRedASR

⨍ Functions 138◇ Types & classes 38
↓ 16 callersMethodget
(self, key, default)
fireredasr/data/token_dict.py:12
↓ 8 callersMethod__init__
(self, idim, n_layers, n_head, d_model, residual_dropout=0.1, dropout_rate=0.1, kernel_size=3
fireredasr/models/module/conformer_encoder.py:7
↓ 8 callersMethodget
(self, key, default)
runtime/triton_tensorrt/model_repo_fireredasr_aed/fireredasr/1/aed_tokenizer.py:15
↓ 5 callersMethod__init__
(self, d_model, n_head, dropout)
fireredasr/models/module/transformer_decoder.py:174
↓ 3 callersFunctioncount_model_parameters
(model)
fireredasr/utils/param.py:6
↓ 3 callersMethodfrom_args
(cls, args)
fireredasr/models/fireredasr_llm.py:30
↓ 3 callersMethodfrom_pretrained
(cls, asr_type, model_dir)
fireredasr/models/fireredasr.py:15
↓ 2 callersMethodbuild_llm_tokenizer
(cls, llm_path, use_flash_attn=False)
fireredasr/tokenizer/llm_tokenizer.py:13
↓ 2 callersMethoddetokenize
inputs is ids or tokens, do not need self.sp
runtime/triton_tensorrt/model_repo_fireredasr_aed/fireredasr/1/aed_tokenizer.py:113
↓ 2 callersMethodforward_attention
(self, attn, v, mask=None)
fireredasr/models/module/conformer_encoder.py:274
↓ 2 callersMethodforward_output
(self, output, residual, sz_b, len_q)
fireredasr/models/module/conformer_encoder.py:254
↓ 2 callersMethodforward_qkv
(self, q, k, v)
fireredasr/models/module/conformer_encoder.py:238
↓ 2 callersMethodfrom_model_dir
(cls, tensorrt_model_dir, tokenizer, device_id=0)
runtime/triton_tensorrt/model_repo_fireredasr_aed/fireredasr/1/fireredasr_aed_tensorrt.py:200
↓ 2 callersMethodgenerate
(self, decoder_input_ids, encoder_outputs, encoder_max_inpu
runtime/triton_tensorrt/model_repo_fireredasr_aed/fireredasr/1/fireredasr_aed_tensorrt.py:141
↓ 2 callersFunctionload_fireredasr_aed_model
(model_path)
fireredasr/models/fireredasr.py:109
↓ 2 callersMethodprint
(self)
fireredasr/utils/wer.py:227
↓ 2 callersFunctionread_uttid2tokens
(filename, do_tn=False, rm_special=False)
fireredasr/utils/wer.py:25
↓ 2 callersFunctionremove_tensor_padding
(input_tensor, input_tensor_lengths=None, pad_value=None)
runtime/triton_tensorrt/model_repo_fireredasr_aed/fireredasr/1/fireredasr_aed_tensorrt.py:61
↓ 2 callersMethodtranscribe
(self, padded_input, input_lengths, beam_size=1, nbest=1, decode_max_len=0,
runtime/triton_tensorrt/model_repo_fireredasr_aed/fireredasr/1/fireredasr_aed_tensorrt.py:228
↓ 1 callersMethod__init__
(self, engine_path, device_id=0, dtype='float16')
runtime/triton_tensorrt/model_repo_fireredasr_aed/fireredasr/1/fireredasr_aed_tensorrt.py:22
↓ 1 callersMethod_merge_input_ids_with_speech_features
Modified from: https://github.com/k2-fsa/icefall/blob/master/egs/speech_llm/ASR_LLM/whisper_llm_zh/model.py
fireredasr/models/fireredasr_llm.py:153
↓ 1 callersMethod_rel_shift
(self, x)
fireredasr/models/module/conformer_encoder.py:302
↓ 1 callersMethodadd
(self, wer_info)
fireredasr/utils/wer.py:224
↓ 1 callersMethodadd
(self, n_en_word, n_en_correct, n_dig_word, n_dig_correct)
fireredasr/utils/wer.py:256
↓ 1 callersMethodbatch_beam_search
(self, encoder_outputs, src_masks, beam_size=1, nbest=1, decode_max_len=0,
fireredasr/models/module/transformer_decoder.py:38
↓ 1 callersMethodclean_text
remove punc, remove space between Chinese and keep space between English
fireredasr/tokenizer/llm_tokenizer.py:24
↓ 1 callersFunctioncompute_one_wer_info
Impl minimum edit distance and backtrace. Args: ref, hyp: List[str] Returns: WerInfo
fireredasr/utils/wer.py:117
↓ 1 callersFunctioncompute_uttid2wer_info
(refs, hyps, print_sentence_wer=False)
fireredasr/utils/wer.py:78
↓ 1 callersFunctionconvert_firered_decoder
(model_args, model_params, quant_algo: str = None)
runtime/triton_tensorrt/scripts/convert_checkpoint.py:122
↓ 1 callersFunctionconvert_onnx_to_trt
Convert an ONNX model to a TensorRT engine.
runtime/triton_tensorrt/scripts/export_encoder_tensorrt.py:150
↓ 1 callersFunctioncount_english_ditgit
(ref, hyp, wer_info)
fireredasr/utils/wer.py:269
↓ 1 callersMethoddetokenize
inputs is ids or tokens, do not need self.sp
fireredasr/tokenizer/aed_tokenizer.py:58
↓ 1 callersFunctionexport_encoder_onnx
Export the conformer encoder model to ONNX format.
runtime/triton_tensorrt/scripts/export_encoder_tensorrt.py:92
↓ 1 callersMethodforward
(self, dec_input, enc_output, self_attn_mask, cross_attn_mask, cache=None)
fireredasr/models/module/transformer_decoder.py:185
↓ 1 callersMethodforward
(self, padded_input, input_lengths)
runtime/triton_tensorrt/model_repo_fireredasr_aed/fireredasr/1/fireredasr_aed_tensorrt.py:34
↓ 1 callersFunctionget_args
()
runtime/triton_tensorrt/infer.py:28
↓ 1 callersFunctionget_decoder_config
(model_args, dtype: str, logits_dtype: str, quant_algo: QuantAlgo)
runtime/triton_tensorrt/scripts/convert_checkpoint.py:45
↓ 1 callersFunctionget_parser
Get the command-line argument parser.
runtime/triton_tensorrt/scripts/export_encoder_tensorrt.py:39
↓ 1 callersMethodget_session
(self, engine_path)
runtime/triton_tensorrt/model_repo_fireredasr_aed/fireredasr/1/fireredasr_aed_tensorrt.py:27
↓ 1 callersMethodget_session
(self, engine_dir, runtime_mapping, debug_mode=False)
runtime/triton_tensorrt/model_repo_fireredasr_aed/fireredasr/1/fireredasr_aed_tensorrt.py:106
↓ 1 callersFunctionget_trt_kwargs_dynamic_batch
Get keyword arguments for TensorRT with dynamic batch size.
runtime/triton_tensorrt/scripts/export_encoder_tensorrt.py:127
↓ 1 callersFunctionget_wav_info
Returns: wavs: list of (uttid, wav_path)
fireredasr/speech2text.py:81
↓ 1 callersMethodget_ys_lengths
(self, ys)
fireredasr/models/module/transformer_decoder.py:166
↓ 1 callersMethodignored_target_position_is_0
(self, padded_targets, ignore_id)
fireredasr/models/module/transformer_decoder.py:142
↓ 1 callersFunctioninit_distributed
()
runtime/triton_tensorrt/infer.py:86
↓ 1 callersMethodload_encoder
(cls, model_path)
fireredasr/models/fireredasr_llm.py:19
↓ 1 callersFunctionload_firered_llm_model_and_tokenizer
(model_path, encoder_path, llm_dir)
fireredasr/models/fireredasr.py:117
↓ 1 callersFunctionmain
(args)
fireredasr/speech2text.py:39
↓ 1 callersFunctionmain
(args)
fireredasr/utils/wer.py:16
↓ 1 callersFunctionmain
()
runtime/triton_tensorrt/infer.py:125
↓ 1 callersFunctionmain
()
runtime/triton_tensorrt/http_client.py:8
↓ 1 callersFunctionmain
Main function to export the model.
runtime/triton_tensorrt/scripts/export_encoder_tensorrt.py:195
↓ 1 callersMethodpad_feat
(self, xs, pad_value)
fireredasr/data/asr_feat.py:32
↓ 1 callersMethodpad_feat
(self, xs, pad_value)
runtime/triton_tensorrt/model_repo_fireredasr_aed/fireredasr/1/asr_feat.py:47
↓ 1 callersMethodpadding_position_is_0
(self, padded_input, input_lengths)
fireredasr/models/module/conformer_encoder.py:45
↓ 1 callersFunctionparse_arguments
()
runtime/triton_tensorrt/scripts/convert_checkpoint.py:15
↓ 1 callersMethodpreprocess_texts
(cls, origin_texts, tokenizer, max_len, decode=False)
fireredasr/tokenizer/llm_tokenizer.py:42
↓ 1 callersFunctionread_config
(component, engine_dir)
runtime/triton_tensorrt/model_repo_fireredasr_aed/fireredasr/1/fireredasr_aed_tensorrt.py:12
↓ 1 callersMethodread_dict
(self, dict_path)
fireredasr/data/token_dict.py:39
↓ 1 callersMethodread_dict
(self, dict_path)
runtime/triton_tensorrt/model_repo_fireredasr_aed/fireredasr/1/aed_tokenizer.py:42
↓ 1 callersMethodread_kaldi_cmvn
(self, kaldi_cmvn_file)
fireredasr/data/asr_feat.py:55
↓ 1 callersMethodread_kaldi_cmvn
(self, kaldi_cmvn_file)
runtime/triton_tensorrt/model_repo_fireredasr_aed/fireredasr/1/asr_feat.py:67
↓ 1 callersFunctionread_uttid2text
(filename, do_tn=False, rm_special=False)
fireredasr/utils/wer.py:35
↓ 1 callersFunctionremap_state_dict
(original_state_dict)
runtime/triton_tensorrt/scripts/convert_checkpoint.py:85
↓ 1 callersMethodset_finished_beam_score_to_zero
(self, scores, is_finished)
fireredasr/models/module/transformer_decoder.py:155
↓ 1 callersMethodset_finished_beam_y_to_eos
(self, ys, is_finished)
fireredasr/models/module/transformer_decoder.py:162
↓ 1 callersFunctiontext2tokens
(text)
fireredasr/utils/wer.py:57
↓ 1 callersMethodtranscribe
(self, batch_uttid, batch_wav_path, args={})
fireredasr/models/fireredasr.py:43
↓ 1 callersMethodupper_triangular_is_0
(self, size)
fireredasr/models/module/transformer_decoder.py:150
Method__call__
(self, wav_paths)
fireredasr/data/asr_feat.py:16
Method__call__
(self, x, is_train=False)
fireredasr/data/asr_feat.py:49
Method__call__
(self, wav, is_train=False)
fireredasr/data/asr_feat.py:88
Method__call__
(self, wavs)
runtime/triton_tensorrt/model_repo_fireredasr_aed/fireredasr/1/asr_feat.py:22
Method__call__
(self, x, is_train=False)
runtime/triton_tensorrt/model_repo_fireredasr_aed/fireredasr/1/asr_feat.py:61
Method__contains__
(self, query)
fireredasr/data/token_dict.py:31
Method__contains__
(self, query)
runtime/triton_tensorrt/model_repo_fireredasr_aed/fireredasr/1/aed_tokenizer.py:34
Method__getitem__
(self, key)
fireredasr/data/token_dict.py:17
Method__getitem__
(self, key)
runtime/triton_tensorrt/model_repo_fireredasr_aed/fireredasr/1/aed_tokenizer.py:20
Method__init__
(self, cost, align)
fireredasr/utils/wer.py:128
Method__init__
(self, ref, err, crt, sub, dele, ins, ali)
fireredasr/utils/wer.py:204
Method__init__
(self)
fireredasr/utils/wer.py:221
Method__init__
(self)
fireredasr/utils/wer.py:250
Method__init__
(self, kaldi_cmvn_file)
fireredasr/data/asr_feat.py:11
Method__init__
(self, kaldi_cmvn_file)
fireredasr/data/asr_feat.py:45
Method__init__
(self, num_mel_bins=80, frame_length=25, frame_shift=10, dither=1.0)
fireredasr/data/asr_feat.py:78
Method__init__
(self, dict_path, unk="")
fireredasr/data/token_dict.py:5
Method__init__
(self, encoder, llm, encoder_projector, freeze_encoder, freeze_llm)
fireredasr/models/fireredasr_llm.py:108
Method__init__
(self, args)
fireredasr/models/fireredasr_aed.py:12
Method__init__
(self, asr_type, feat_extractor, model, tokenizer)
fireredasr/models/fireredasr.py:36
Method__init__
(self, d_model, n_head, residual_dropout=0.1, dropout_rate=0.1, kernel_size=
fireredasr/models/module/conformer_encoder.py:55
Method__init__
(self, idim, d_model, out_channels=32)
fireredasr/models/module/conformer_encoder.py:82
Method__init__
(self, d_model, max_len=5000)
fireredasr/models/module/conformer_encoder.py:121
Method__init__
(self, d_model, dropout_rate=0.1)
fireredasr/models/module/conformer_encoder.py:146
Method__init__
(self, d_model, kernel_size=33, dropout_rate=0.1)
fireredasr/models/module/conformer_encoder.py:169
Method__init__
(self, n_head, d_model, residual_dropout=0.1)
fireredasr/models/module/conformer_encoder.py:207
Method__init__
(self, temperature)
fireredasr/models/module/conformer_encoder.py:263
Method__init__
(self, n_head, d_model, residual_dropout=0.1)
fireredasr/models/module/conformer_encoder.py:290
Method__init__
(self, encoder_dim, llm_dim, downsample_rate=2)
fireredasr/models/module/adapter.py:6
Method__init__
( self, sos_id, eos_id, pad_id, odim, n_layers, n_head, d_model, residual_
fireredasr/models/module/transformer_decoder.py:10
Method__init__
(self, d_model, n_head, dropout=0.1)
fireredasr/models/module/transformer_decoder.py:215
next →1–100 of 138, ranked by callers