MCPcopy Create free account

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

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

↓ 2 callersMethodposition_encoding
For getting encoding in a streaming fashion Attention!!!!! we apply dropout only once at the whole utterance level in a none
OSUM-EChat/tts/cosyvoice/transformer/embedding.py:79
↓ 2 callersMethodposition_encoding
(self, offset: Union[int, torch.Tensor], size: int)
OSUM-EChat/wenet/transformer/subsampling.py:32
↓ 2 callersMethodposition_encoding
For getting encoding in a streaming fashion Attention!!!!! we apply dropout only once at the whole utterance level in a none
OSUM-EChat/wenet/transformer/embedding.py:80
↓ 2 callersMethodposition_encoding
For getting encoding in a streaming fashion Attention!!!!! we apply dropout only once at the whole utterance level in a none
OSUM/wenet/transformer/embedding.py:80
↓ 2 callersFunctionpost_process
(list_out, file_obj, keys)
OSUM/tools/onnx2horizonbin.py:236
↓ 2 callersMethodprefix_beam_search
prefix beam search also see wenet.transducer.transducer.beam_search
OSUM/wenet/transducer/search/prefix_beam_search.py:42
↓ 2 callersFunctionprint_result
(info)
OSUM/tools/websocket/performance-ws.py:98
↓ 2 callersFunctionread_lists
(list_file)
OSUM-EChat/tts/cosyvoice/utils/file_utils.py:24
↓ 2 callersFunctionread_mat
[mat] = read_mat(file_or_fd) Reads single kaldi matrix, supports ascii and binary. file_or_fd : file, gzipped file, pipe or opened file descrip
OSUM/wenet/dataset/kaldi_io.py:432
↓ 2 callersFunctionread_non_lang_symbols
read non-linguistic symbol from file. The file format is like below: {NOISE}\n {BRK}\n ... Args: non_lang_sym_path: no
OSUM-EChat/wenet/utils/file_utils.py:26
↓ 2 callersFunctionread_non_lang_symbols
read non-linguistic symbol from file. The file format is like below: {NOISE}\n {BRK}\n ... Args: non_lang_sym_path: no
OSUM/wenet/utils/file_utils.py:26
↓ 2 callersFunctionread_vec_flt
[flt-vec] = read_vec_flt(file_or_fd) Read kaldi float vector, ascii or binary input,
OSUM/wenet/dataset/kaldi_io.py:306
↓ 2 callersFunctionread_vec_int
[int-vec] = read_vec_int(file_or_fd) Read kaldi integer vector, ascii or binary input,
OSUM/wenet/dataset/kaldi_io.py:195
↓ 2 callersFunctionreinit_lora_modules
r"""Refer to https://github.com/Outsider565/LoRA-GA/blob/ c185846309ea9012d0bcd46ebd30347dda1c592c/run_exp.py#L67 Reinitialize the lora model
OSUM/wenet/finetune/lora/utils.py:157
↓ 2 callersFunctionremove_duplicates_and_blank
(hyp: List[int], blank_id: int = 0)
OSUM/wenet/utils/ctc_utils.py:23
↓ 2 callersMethodsample
Sample data according to rank/world_size/num_workers Args: data(List): input data list Returns:
OSUM-EChat/tts/cosyvoice/dataset/dataset.py:83
↓ 2 callersFunctionsave_checkpoint
Args: infos (dict or None): any info you want to save.
OSUM-EChat/wenet/utils/checkpoint.py:73
↓ 2 callersFunctionsave_checkpoint
Args: infos (dict or None): any info you want to save.
OSUM/wenet/utils/checkpoint.py:62
↓ 2 callersFunctionsave_model
(model, model_name, info_dict)
OSUM-EChat/tts/cosyvoice/utils/train_utils.py:195
↓ 2 callersFunctionsave_model
(model, info_dict)
OSUM-EChat/wenet/utils/train_utils.py:627
↓ 2 callersFunctionsave_model
(model, info_dict)
OSUM/wenet/utils/train_utils.py:621
↓ 2 callersFunctionsave_state_dict_and_infos
(state_dict, path: str, infos=None)
OSUM-EChat/wenet/utils/checkpoint.py:59
↓ 2 callersFunctionsave_state_dict_and_infos
(state_dict, path: str, infos=None)
OSUM/wenet/utils/checkpoint.py:48
↓ 2 callersMethodset_epoch
(self, epoch)
OSUM-EChat/tts/cosyvoice/dataset/dataset.py:114
↓ 2 callersMethodset_epoch
(self, epoch)
OSUM/wenet/dataset/dataset.py:137
↓ 2 callersMethodset_step
(self, step: int)
OSUM-EChat/tts/cosyvoice/utils/scheduler.py:70
↓ 2 callersMethodsolve_euler
Fixed euler solver for ODEs. Args: x (torch.Tensor): random noise t_span (torch.Tensor): n_timesteps interpol
OSUM-EChat/tts/cosyvoice/flow/flow_matching.py:71
↓ 2 callersMethodsos_symbol
Export interface for c++ call, return sos symbol id of the model
OSUM/wenet/transformer/asr_model.py:402
↓ 2 callersMethodsplit_data
(self, total_num)
OSUM-EChat/wenet/dataset/dataset.py:85
↓ 2 callersFunctionsplit_paragraph
(text: str, tokenize, lang="zh", token_max_n=80, token_min_n=60, merge_len=20, comma_split=False)
OSUM-EChat/tts/cosyvoice/utils/frontend_utils.py:65
↓ 2 callersFunctionstatic_batch
Static batch the data by `batch_size` Args: data: Iterable[{key, feat, label}] batch_size: batch size Retur
OSUM-EChat/tts/cosyvoice/dataset/processor.py:296
↓ 2 callersMethodsubsampling_rate
Export interface for c++ call, return subsampling_rate of the model
OSUM/wenet/transformer/asr_model.py:389
↓ 2 callersFunctionsubsequent_chunk_mask
Create mask for subsequent steps (size, size) with chunk size, this is for streaming encoder Args: size (int): size of mask
OSUM-EChat/tts/cosyvoice/utils/mask.py:129
↓ 2 callersFunctionsubsequent_chunk_mask
Create mask for subsequent steps (size, size) with chunk size, this is for streaming encoder Args: size (int): size of mask
OSUM-EChat/wenet/utils/mask.py:88
↓ 2 callersFunctionsubsequent_chunk_mask
Create mask for subsequent steps (size, size) with chunk size, this is for streaming encoder Args: size (int): size of mask
OSUM/wenet/utils/mask.py:88
↓ 2 callersFunctionsubsequent_mask
Create mask for subsequent steps (size, size). This mask is used only in decoder which works in an auto-regressive mode. This means the curre
OSUM-EChat/wenet/utils/mask.py:52
↓ 2 callersMethodtar_file_and_group
(self, sample)
OSUM/tools/extract_shard_data.py:70
↓ 2 callersMethodtie_or_clone_weights
Tie or clone module weights (between word_emb and output_layer) depending of whether we are using TorchScript or not
OSUM-EChat/tts/cosyvoice/transformer/decoder.py:230
↓ 2 callersMethodtie_or_clone_weights
Tie or clone module weights (between word_emb and output_layer) depending of whether we are using TorchScript or not
OSUM-EChat/wenet/transformer/decoder.py:283
↓ 2 callersMethodtie_or_clone_weights
Tie or clone module weights (between word_emb and output_layer) depending of whether we are using TorchScript or not
OSUM/wenet/transformer/decoder.py:283
↓ 2 callersFunctiontokenize
Decode text to chars or BPE Inplace operation Args: data: Iterable[{key, wav, txt, sample_rate}] Returns:
OSUM-EChat/tts/cosyvoice/dataset/processor.py:225
↓ 2 callersFunctiontokenize_by_bpe_model
(sp, txt)
OSUM-EChat/wenet/text/tokenize_utils.py:18
↓ 2 callersFunctiontokenize_by_bpe_model
(sp, txt)
OSUM/wenet/text/tokenize_utils.py:18
↓ 2 callersMethodtotal_score
(self)
OSUM-EChat/wenet/transformer/search.py:92
↓ 2 callersMethodtotal_score
(self)
OSUM/wenet/transformer/search.py:92
↓ 2 callersFunctiontpr_loss
(disc_real_outputs, disc_generated_outputs, tau)
OSUM-EChat/tts/cosyvoice/utils/losses.py:5
↓ 2 callersMethodtts_gxl
(self, text, flow_embedding, llm_embedding=torch.zeros(0, 192), prompt_text=torch.zeros(1, 0, dtyp
OSUM-EChat/tts/cosyvoice/cli/model.py:249
↓ 2 callersMethodupdate_context
(self, context_graph, prefix_score, word_id)
OSUM-EChat/wenet/transformer/search.py:99
↓ 2 callersMethodupdate_context
(self, context_graph, prefix_score, word_id)
OSUM/wenet/transformer/search.py:99
↓ 2 callersFunctionwidth
(string)
OSUM/tools/compute-cer.py:250
↓ 2 callersFunctionwidth
(string)
OSUM/tools/compute-wer.py:243
↓ 1 callersFunctionBigDataset
Construct dataset from arguments We have two shuffle stage in the Dataset. The first is global shuffle at shard tar/raw file level.
OSUM-EChat/wenet/dataset/dataset.py:338
↓ 1 callersMethodCTL
(self, x, y, negs, mask=None)
OSUM/wenet/ctl_model/asr_model_ctl.py:260
↓ 1 callersFunctionDataset
Construct dataset from arguments We have two shuffle stage in the Dataset. The first is global shuffle at shards tar/raw file level.
OSUM-EChat/common_utils/fake_data/combine/dataset/dataset_no_wav.py:123
↓ 1 callersFunctionDataset
Construct dataset from arguments for ssl model We have two shuffle stage in the Dataset. The first is global shuffle at shards tar/r
OSUM/wenet/ssl/init_dataset.py:42
↓ 1 callersMethod__init__
(self, input_size: int = 512, output_size: int = 80, spk_em
OSUM-EChat/tts/cosyvoice/flow/flow.py:25
↓ 1 callersMethod__init__
(self, in_channels, cfm_params, n_spks=1, spk_emb_dim=64, estimator: torch.nn.Module = None)
OSUM-EChat/tts/cosyvoice/flow/flow_matching.py:21
↓ 1 callersMethod__init__
( self, n_expert: int, n_expert_per_token: int, idim: int,
OSUM-EChat/tts/cosyvoice/transformer/positionwise_feed_forward.py:75
↓ 1 callersMethod__init__
( self, vocab_size: int, encoder_output_size: int, attention_heads: int = 4,
OSUM-EChat/tts/cosyvoice/transformer/decoder.py:58
↓ 1 callersMethod__init__
Construct an MultiHeadedAttention object.
OSUM-EChat/tts/cosyvoice/transformer/attention.py:36
↓ 1 callersMethod__init__
Construct an EncoderLayer object.
OSUM-EChat/tts/cosyvoice/transformer/encoder_layer.py:129
↓ 1 callersMethod__init__
( self, vocab_size: int, encoder_output_size: int, attention_heads: int = 4,
OSUM-EChat/wenet/transformer/decoder.py:63
↓ 1 callersMethod__init__
Construct CTC module Args: odim: dimension of outputs encoder_output_size: number of encoder projection units
OSUM-EChat/wenet/transformer/ctc.py:73
↓ 1 callersMethod__init__
Construct an EncoderLayer object.
OSUM-EChat/wenet/transformer/encoder_layer.py:150
↓ 1 callersMethod__init__
( self, size: int, linear_units: int, kernel_size: int, dropout_rate:
OSUM/wenet/branchformer/cgmlp.py:137
↓ 1 callersMethod__init__
( self, input_size: int, output_size: int = 256, use_attn: bool = True,
OSUM/wenet/branchformer/encoder.py:32
↓ 1 callersMethod__init__
( self, idim, l_order, r_order, threshold=1.0, dropout=0.1,
OSUM/wenet/paraformer/paraformer.py:39
↓ 1 callersMethod__init__
( self, idim, l_order, r_order, threshold=1.0, dropout=0.1,
OSUM/wenet/paraformer/cif.py:26
↓ 1 callersMethod__init__
( self, vocab_size: int, encoder_output_size: int, attention_heads: int = 4,
OSUM/wenet/transformer/decoder.py:63
↓ 1 callersMethod__init__
Construct an EncoderLayer object.
OSUM/wenet/transformer/encoder_layer.py:150
↓ 1 callersMethod__init__
Construct DualConformerEncoder Support both the full context mode and the streaming mode separately
OSUM/wenet/ctl_model/encoder.py:99
↓ 1 callersMethod_apply_mask
( self, xs: torch.Tensor, xs_masks: torch.Tensor)
OSUM/wenet/ssl/w2vbert/w2vbert_model.py:253
↓ 1 callersMethod_apply_mask
( self, xs: torch.Tensor, xs_masks: torch.Tensor)
OSUM/wenet/ssl/wav2vec2/wav2vec2_model.py:285
↓ 1 callersMethod_apply_mask_signal
( self, input: torch.Tensor, input_lens: torch.Tensor)
OSUM/wenet/ssl/bestrq/bestrq_model.py:216
↓ 1 callersFunction_batch_beam_search
Perform batch beam search Args: logit: shape (batch_size, seq_length, vocab_size) masks: shape (batch_size, seq_leng
OSUM/wenet/paraformer/search.py:200
↓ 1 callersMethod_cal_attn_score
( self, encoder_out: torch.Tensor, encoder_mask: torch.Tensor, hyps_pad: torch
OSUM/wenet/transducer/transducer.py:187
↓ 1 callersMethod_cal_transducer_score
( self, encoder_out: torch.Tensor, encoder_mask: torch.Tensor, hyps_lens: torc
OSUM/wenet/transducer/transducer.py:160
↓ 1 callersMethod_calc_att_loss
( self, encoder_out: torch.Tensor, encoder_mask: torch.Tensor, ys_pad: torch.T
OSUM-EChat/wenet/transformer/asr_model.py:236
↓ 1 callersMethod_calc_att_loss
( self, encoder_out: torch.Tensor, encoder_mask: torch.Tensor, ys_pad: torch.T
OSUM/wenet/paraformer/paraformer.py:218
↓ 1 callersMethod_calc_lfmmi_loss
(self, encoder_out, encoder_mask, text)
OSUM/wenet/k2/model.py:100
↓ 1 callersMethod_compute_loss
(self, encoder_out: torch.Tensor, encoder_out_lens: torch.Tensor,
OSUM/wenet/transducer/transducer.py:474
↓ 1 callersMethod_compute_loss
(self, input: torch.Tensor, target: torch.Tensor, mask: torch.Tensor)
OSUM/wenet/ssl/bestrq/bestrq_model.py:279
↓ 1 callersMethod_compute_mlm_loss
(self, input: torch.Tensor, target: torch.Tensor, mask: torch.Tensor)
OSUM/wenet/ssl/w2vbert/w2vbert_model.py:273
↓ 1 callersFunction_cosine_annealing
(initial_lr, step, max_steps, min_lr)
OSUM-EChat/tts/cosyvoice/utils/scheduler.py:391
↓ 1 callersFunction_cosine_annealing
(initial_lr, step, max_steps, min_lr)
OSUM-EChat/wenet/utils/scheduler.py:396
↓ 1 callersFunction_cosine_annealing
(initial_lr, step, max_steps, min_lr)
OSUM/wenet/utils/scheduler.py:396
↓ 1 callersMethod_extract_text_token_generator
(self, text_generator)
OSUM-EChat/tts/cosyvoice/cli/frontend.py:95
↓ 1 callersMethod_f02uv
(self, f0)
OSUM-EChat/tts/cosyvoice/hifigan/generator.py:133
↓ 1 callersMethod_fill_fail_output
This function fills the fail arc for each trie node, it can be computed in linear time by performing a breadth-first search starting from the
OSUM-EChat/wenet/utils/context_graph.py:176
↓ 1 callersMethod_fill_fail_output
This function fills the fail arc for each trie node, it can be computed in linear time by performing a breadth-first search starting from the
OSUM/wenet/utils/context_graph.py:176
↓ 1 callersFunction_filter_dict_fields
(input_dict, fields_to_keep)
OSUM/wenet/paraformer/convert_paraformer_to_wenet_config_and_ckpt.py:46
↓ 1 callersMethod_forward
( self, x: torch.Tensor, mask: torch.Tensor, pos_emb: torch.Tensor, ma
OSUM/wenet/branchformer/encoder_layer.py:111
↓ 1 callersMethod_forward
( self, x: torch.Tensor, mask: torch.Tensor, pos_emb: torch.Tensor, ma
OSUM/wenet/e_branchformer/encoder_layer.py:93
↓ 1 callersMethod_forward_ctc
( self, encoder_out: torch.Tensor, encoder_mask: torch.Tensor, text: torch.Tensor,
OSUM/wenet/k2/model.py:60
↓ 1 callersMethod_forward_encoder
( self, speech: torch.Tensor, speech_lengths: torch.Tensor, decoding_chunk_siz
OSUM-EChat/wenet/transformer/asr_model.py:271
↓ 1 callersMethod_forward_encoder_blocks
( self, xs: torch.Tensor, xs_masks: torch.Tensor, pos_emb: torch.Tensor, mask_pad: torch.Tenso
OSUM/wenet/ssl/w2vbert/w2vbert_model.py:296
↓ 1 callersMethod_forward_encoder_blocks
(self, xs: torch.Tensor, xs_masks: torch.Tensor, pos_emb: torch.Tensor, mask_p
OSUM/wenet/ssl/wav2vec2/wav2vec2_model.py:312
↓ 1 callersMethod_forward_subsampling
( self, xs: torch.Tensor, xs_lens: torch.Tensor )
OSUM/wenet/ssl/w2vbert/w2vbert_model.py:286
↓ 1 callersMethod_forward_subsampling
( self, xs: torch.Tensor, xs_lens: torch.Tensor )
OSUM/wenet/ssl/wav2vec2/wav2vec2_model.py:302
↓ 1 callersMethod_get_constant_lr
(self, step)
OSUM-EChat/tts/cosyvoice/utils/scheduler.py:369
← previousnext →301–400 of 1,852, ranked by callers