MCPcopy Create free account

hub / github.com/ASLP-lab/OSUM / functions

Functions1,852 in github.com/ASLP-lab/OSUM

↓ 2 callersMethod__init__
( self, idim: int, hidden_units: int, dropout_rate: float, activation:
OSUM-EChat/wenet/transformer/positionwise_feed_forward.py:78
↓ 2 callersMethod__init__
(self, n_head, in_feat, n_feat, dropout_ra
OSUM/wenet/paraformer/attention.py:121
↓ 2 callersMethod__init__
( self, idim: int, hidden_units: int, dropout_rate: float, activation:
OSUM/wenet/transformer/positionwise_feed_forward.py:78
↓ 2 callersMethod__init__
Args: input_size (int): input dim output_size (int): dimension of attention attention_heads (int): the nu
OSUM/wenet/transformer/encoder.py:40
↓ 2 callersMethod_add_bos_eos
(self, bos, eos, inputs_embeds, attention_mask, target=None)
OSUM/wenet/llm_asr/llmasr_model.py:313
↓ 2 callersMethod_build_sp
(self)
OSUM-EChat/wenet/text/bpe_tokenizer.py:25
↓ 2 callersMethod_build_sp
(self)
OSUM/wenet/text/bpe_tokenizer.py:25
↓ 2 callersMethod_calc_att_loss
( self, encoder_out: torch.Tensor, encoder_mask: torch.Tensor, ys_pad: torch.T
OSUM/wenet/whisper/whisper.py:63
↓ 2 callersMethod_calc_att_loss
( self, encoder_out: torch.Tensor, encoder_mask: torch.Tensor, ys_pad: torch.T
OSUM/wenet/transformer/asr_model.py:236
↓ 2 callersFunction_compute_contrastive_loss
(quantized_features: torch.Tensor, features: torch.Tensor,
OSUM/wenet/ssl/wav2vec2/wav2vec2_model.py:58
↓ 2 callersMethod_compute_perplexity
(probs, mask=None)
OSUM/wenet/ssl/wav2vec2/quantizer.py:55
↓ 2 callersMethod_conv_forward
(self, input: Tensor, weight: Tensor, bias: Optional[Tensor])
OSUM/wenet/squeezeformer/conv2d.py:53
↓ 2 callersMethod_decode
(self, audio_file: str, tokens_info: bool = False, label: str
OSUM/wenet/cli/model.py:84
↓ 2 callersMethod_forward_paraformer
( self, speech: torch.Tensor, speech_lengths: torch.Tensor, decoding_chunk_siz
OSUM/wenet/paraformer/paraformer.py:326
↓ 2 callersMethod_get_embedding_for_history
prompt_patern1,prompt,history, wav, prompt_patern2,txt,answer_wav, historcy_batch的内容是: [ big_embed, [
OSUM-EChat/wenet/osum_echat/llmasr_model_instruct_version.py:225
↓ 2 callersMethod_get_lr
Simple const lr policy
OSUM-EChat/tts/cosyvoice/utils/scheduler.py:132
↓ 2 callersMethod_get_lr
Simple const lr policy
OSUM-EChat/wenet/utils/scheduler.py:137
↓ 2 callersMethod_get_lr
Simple const lr policy
OSUM/wenet/utils/scheduler.py:137
↓ 2 callersFunction_isAllAlpha
(word: Union[List[Any], str])
OSUM/wenet/paraformer/search.py:35
↓ 2 callersFunction_isAllChinese
(word: Union[List[Any], str])
OSUM/wenet/paraformer/search.py:16
↓ 2 callersFunction_isChinese
(ch: str)
OSUM/wenet/paraformer/search.py:10
↓ 2 callersFunction_sample_negative_indices
Sample `num_negatives` vectors from feature vectors.
OSUM/wenet/ssl/wav2vec2/wav2vec2_model.py:15
↓ 2 callersFunction_tokenize_by_seg_dic_or_bpe_model
( txt, sp=None, seg_dict=None, upper=True, )
OSUM-EChat/wenet/text/tokenize_utils.py:28
↓ 2 callersFunction_tokenize_by_seg_dic_or_bpe_model
( txt, sp=None, seg_dict=None, upper=True, )
OSUM/wenet/text/tokenize_utils.py:28
↓ 2 callersFunctionadd_blank
Prepad blank for transducer predictor Args: ys_pad (torch.Tensor): batch of padded target sequences (B, Lmax) blank (int): index
OSUM/wenet/utils/common.py:79
↓ 2 callersFunctionadd_optional_chunk_mask
Apply optional mask for encoder. Args: xs (torch.Tensor): padded input, (B, L, D), L for max length mask (torch.Tensor): mask fo
OSUM-EChat/wenet/utils/mask.py:126
↓ 2 callersFunctionattention_rescoring
Args: ctc_prefix_results(List[DecodeResult]): ctc prefix beam search results
OSUM/wenet/transformer/search.py:369
↓ 2 callersFunctionbatch_forward
(model, batch, scaler, info_dict, device)
OSUM-EChat/wenet/utils/train_utils.py:688
↓ 2 callersFunctionbatch_forward
(model, batch, scaler, info_dict, device)
OSUM/wenet/utils/train_utils.py:682
↓ 2 callersFunctioncalc_utt_length
(_text: str)
OSUM-EChat/tts/cosyvoice/utils/frontend_utils.py:66
↓ 2 callersFunctioncharacterize
(string)
OSUM/tools/compute-cer.py:16
↓ 2 callersFunctioncharacterize
(string)
OSUM/tools/compute-wer.py:15
↓ 2 callersFunctioncheck_txt_format
(s)
OSUM-EChat/wenet/dataset/process/processor.py:86
↓ 2 callersFunctioncheck_wav_format
(s)
OSUM-EChat/wenet/dataset/process/processor.py:79
↓ 2 callersMethodcluster
(self, data)
OSUM/tools/compute-cer.py:235
↓ 2 callersMethodcluster
(self, data)
OSUM/tools/compute-wer.py:228
↓ 2 callersFunctioncontain_oov
(units)
OSUM/tools/fst/prepare_dict.py:18
↓ 2 callersFunctionconvert_numbers_in_string
(s)
OSUM-EChat/common_utils/utils4infer.py:113
↓ 2 callersFunctioncosyvoice_join
(group_join, info_dict)
OSUM-EChat/tts/cosyvoice/utils/train_utils.py:217
↓ 2 callersFunctionctc_greedy_search
(ctc_probs: torch.Tensor, ctc_lens: torch.Tensor, blank_id: int =
OSUM/wenet/transformer/search.py:107
↓ 2 callersMethodctc_logprobs
(self, encoder_out: torch.Tensor, blank_penalty: float = 0.0,
OSUM/wenet/transformer/asr_model.py:297
↓ 2 callersFunctionctc_prefix_beam_search
Returns: List[List[List[int]]]: nbest result for each utterance
OSUM-EChat/wenet/transformer/search.py:125
↓ 2 callersMethodcv
Cross validation on
OSUM/wenet/utils/executor.py:139
↓ 2 callersMethoddecode
(self, x: torch.Tensor, s: torch.Tensor = torch.zeros(1, 1, 0))
OSUM-EChat/tts/cosyvoice/hifigan/generator.py:350
↓ 2 callersMethoddetokenize
(self, ids: List[int])
OSUM/wenet/text/base_tokenizer.py:14
↓ 2 callersFunctiondo_compute_log_mel_spectrogram
(waveform,n_fft=400, hop_length=160, num_mel_b
OSUM-EChat/wenet/dataset/process/processor.py:1157
↓ 2 callersFunctiondo_decode
(input_wav_path, input_prompt)
OSUM/infer_gradio.py:113
↓ 2 callersFunctiondo_get_first_tag_from_str
得到第一个<>标签的内容,并将其转为大写 >>> input : "<tag1>content1<tag2>content2<tag3>content3" >>> output: "<TAG1>" :param upper: if false, return the
OSUM/tools/compute-acc.py:17
↓ 2 callersFunctiondo_s2t_speech_understanding
(model, input_wav_path, input_prompt)
OSUM-EChat/infer_runtime.py:42
↓ 2 callersFunctionestimate_gradient
r""" Estimate the gradient of the model on the given dataset
OSUM/wenet/finetune/lora/utils.py:116
↓ 2 callersFunctionexport_ctc
(asr_model, args)
OSUM/wenet/bin/export_onnx_bpu.py:976
↓ 2 callersFunctionexport_encoder
(asr_model, args)
OSUM/wenet/bin/export_onnx_bpu.py:855
↓ 2 callersMethodextend_pe
Reset the positional encodings.
OSUM-EChat/tts/cosyvoice/transformer/embedding.py:224
↓ 2 callersMethodfilter_blank_embedding
( self, ctc_probs: torch.Tensor, encoder_out: torch.Tensor)
OSUM-EChat/wenet/transformer/asr_model.py:208
↓ 2 callersMethodfilter_blank_embedding
( self, ctc_probs: torch.Tensor, encoder_out: torch.Tensor)
OSUM/wenet/transformer/asr_model.py:208
↓ 2 callersFunctionforce_align
ctc forced alignment. Args: torch.Tensor ctc_probs: hidden state sequence, 2d tensor (T, D) torch.Tensor y: id sequence tensor 1d
OSUM/wenet/utils/ctc_utils.py:106
↓ 2 callersMethodforward
(self, x: torch.Tensor)
OSUM/wenet/finetune/lora/layers.py:164
↓ 2 callersMethodforward_chunk
Forward just one chunk Args: xs (torch.Tensor): chunk input, with shape (b=1, time, mel-dim), where `time == (ch
OSUM-EChat/tts/cosyvoice/transformer/encoder.py:184
↓ 2 callersMethodforward_chunk
Forward just one chunk Args: xs (torch.Tensor): chunk input, with shape (b=1, time, mel-dim), where `time == (ch
OSUM-EChat/wenet/transformer/encoder.py:208
↓ 2 callersMethodforward_cif_peaks
(self, alphas: torch.Tensor, token_nums: torch.Tensor)
OSUM/wenet/paraformer/paraformer.py:317
↓ 2 callersMethodforward_layers
(self, xs: torch.Tensor, chunk_masks: torch.Tensor, pos_emb: torch.Tensor,
OSUM-EChat/wenet/transformer/encoder.py:187
↓ 2 callersMethodforward_layers_checkpointed
(self, xs: torch.Tensor, chunk_masks: torch.Tensor,
OSUM-EChat/wenet/transformer/encoder.py:195
↓ 2 callersMethodforward_qkv
Transform query, key and value. Args: query (torch.Tensor): Query tensor (#batch, time1, size). key (torch.Tensor): K
OSUM-EChat/tts/cosyvoice/transformer/attention.py:53
↓ 2 callersMethodfrontend_sft
(self, tts_text, spk_id)
OSUM-EChat/tts/cosyvoice/cli/frontend.py:160
↓ 2 callersMethodgenerate4chat
( self, wavs, wavs_len, prompt=" ", do_sample=True
OSUM-EChat/wenet/osum_echat/llmasr_model_instruct_version.py:759
↓ 2 callersMethodgenerate4chat_think
( self, wavs, wavs_len, prompt=" ", do_sample=True
OSUM-EChat/wenet/osum_echat/llmasr_model_instruct_version.py:825
↓ 2 callersMethodgenerate_s2s_no_stream_think_with_repetition_penalty
( self, wavs, wavs_len, )
OSUM-EChat/wenet/osum_echat/llmasr_model_instruct_version.py:1096
↓ 2 callersMethodgenerate_s2s_no_stream_with_repetition_penalty
( self, wavs, wavs_len, )
OSUM-EChat/wenet/osum_echat/llmasr_model_instruct_version.py:1026
↓ 2 callersMethodgenerate_text2text
( self, device, text, )
OSUM-EChat/wenet/osum_echat/llmasr_model_instruct_version.py:960
↓ 2 callersMethodgenerate_tts
( self, device, text, )
OSUM-EChat/wenet/osum_echat/llmasr_model_instruct_version.py:891
↓ 2 callersFunctionget_args
()
OSUM/wenet/bin/export_onnx_cpu.py:38
↓ 2 callersFunctionget_dummy_input
(batch_size, seq_len, out_channels, device)
OSUM-EChat/tts/cosyvoice/bin/export_onnx.py:33
↓ 2 callersMethodget_model_by_lang
(lang: str)
OSUM/wenet/cli/hub.py:87
↓ 2 callersFunctionget_model_type
(configs)
OSUM-EChat/tts/cosyvoice/utils/class_utils.py:77
↓ 2 callersFunctionget_padding
(kernel_size, dilation=1)
OSUM-EChat/tts/cosyvoice/utils/common.py:98
↓ 2 callersFunctionget_record_gradient_hook
(model, record_dict)
OSUM/wenet/finetune/lora/utils.py:102
↓ 2 callersMethodget_speech_token_label_embedding
(self, speech_token_labels, speech_tokens_length)
OSUM-EChat/wenet/osum_echat/llmasr_model_instruct_version.py:211
↓ 2 callersFunctionget_subsample
(config)
OSUM/wenet/utils/common.py:294
↓ 2 callersFunctionget_wav_from_token_list
(input_list, prompt_speech)
OSUM-EChat/infer_gradio.py:250
↓ 2 callersMethodidentity
(self, x)
OSUM/wenet/finetune/lora/layers.py:36
↓ 2 callersMethodinit_bs
(self)
OSUM/wenet/transducer/transducer.py:155
↓ 2 callersFunctioninit_dataset
(dataset_type, data_type, data_list_file, tokenizer: Option
OSUM-EChat/wenet/utils/init_dataset.py:25
↓ 2 callersFunctioninit_model
(args, configs)
OSUM-EChat/wenet/utils/init_model.py:99
↓ 2 callersFunctioninsert_at_position
将 item_str 插入到 lst 的第 position 个位置(1-based), 若 lst 长度不足则以 "-1" 填充至目标长度后再插入。
OSUM-EChat/wenet/dataset/process/processor.py:60
↓ 2 callersFunctionlexicon_to_fst_no_sil
Convert a lexicon to an FST (in k2 format). Args: lexicon: The input lexicon. See also :func:`read_lexicon` token2id:
OSUM/tools/k2/prepare_char.py:46
↓ 2 callersFunctionload_cmvn
(cmvn_file, is_json)
OSUM/wenet/utils/cmvn.py:88
↓ 2 callersMethodload_hlg_resource_if_necessary
(self, hlg, word)
OSUM/wenet/k2/model.py:123
↓ 2 callersMethodload_jit
(self, llm_text_encoder_model, llm_llm_model, flow_encoder_model)
OSUM-EChat/tts/cosyvoice/cli/model.py:88
↓ 2 callersFunctionload_trained_modules
(model: torch.nn.Module, args: None)
OSUM-EChat/wenet/utils/checkpoint.py:109
↓ 2 callersFunctionload_trained_modules
(model: torch.nn.Module, args: None)
OSUM/wenet/utils/checkpoint.py:98
↓ 2 callersMethodload_trt
(self, flow_decoder_estimator_model, flow_decoder_onnx_model, fp16)
OSUM-EChat/tts/cosyvoice/cli/model.py:96
↓ 2 callersFunctionmake_poly_distortion
Generate a db-domain ploynomial distortion function f(x) = a * x^m * (1-x)^n + x Args: conf: a dict {'a': #int, 'm': #int, 'n':
OSUM/wenet/dataset/wav_distortion.py:31
↓ 2 callersFunctionnormalize
sentence, ignore_words are both in unicode
OSUM/tools/compute-cer.py:67
↓ 2 callersFunctionnormalize
sentence, ignore_words are both in unicode
OSUM/tools/compute-wer.py:64
↓ 2 callersFunctionol
(n)
OSUM/tools/fst/ctc_token_fst_corrected.py:10
↓ 2 callersMethodoutput_size
(self)
OSUM-EChat/wenet/transformer/encoder.py:121
↓ 2 callersMethodoutput_size
(self)
OSUM/wenet/transducer/predictor.py:90
↓ 2 callersMethodoutput_size
(self)
OSUM/wenet/efficient_conformer/encoder.py:234
↓ 2 callersFunctionparaformer_greedy_search
( decoder_out: torch.Tensor, decoder_out_lens: torch.Tensor, cif_peaks: Optional[torch
OSUM/wenet/paraformer/search.py:139
↓ 2 callersFunctionpopen
(cmd, mode="rb")
OSUM/wenet/dataset/kaldi_io.py:89
← previousnext →201–300 of 1,852, ranked by callers