MCPcopy Create free account

hub / github.com/0nutation/SpeechGPT / functions

Functions39 in github.com/0nutation/SpeechGPT

⨍ Functions 39◇ Types & classes 15
↓ 5 callersFunctionextract_text_between_tags
(text, tag1='[SpeechGPT] :', tag2='<eoa>')
speechgpt/src/infer/cli_infer.py:39
↓ 4 callersMethodgenerate_prompt
( self, prefix: str, text: Union[None, str] = None, )
speechgpt/utils/prompter.py:13
↓ 2 callersMethodforward
( self, prompts: List[str] )
speechgpt/src/infer/cli_infer.py:140
↓ 2 callersMethodpostprocess
( self, input )
speechgpt/utils/text2unit/text2unit.py:44
↓ 2 callersFunctiontokenize
(prompt, add_eos_token=True)
speechgpt/src/train/com_sft.py:241
↓ 2 callersFunctiontokenize
(prompt, add_eos_token=True)
speechgpt/src/train/cm_sft.py:179
↓ 1 callersMethod__init__
(self, km_path)
speechgpt/utils/speech2unit/speech2unit.py:91
↓ 1 callersFunctioncli_main
()
speechgpt/utils/vocoder/generate_waveform_from_code.py:82
↓ 1 callersFunctiondump_result
(args, sample_id, pred_wav, suffix="")
speechgpt/utils/vocoder/generate_waveform_from_code.py:26
↓ 1 callersMethoddump_wav
(self, sample_id, pred_wav, prefix)
speechgpt/src/infer/cli_infer.py:209
↓ 1 callersMethodforward
( self, text, **kwargs )
speechgpt/utils/text2unit/text2unit.py:34
↓ 1 callersMethodget_feats
(self, waveform)
speechgpt/utils/speech2unit/speech2unit.py:61
↓ 1 callersMethodinteract
(self)
speechgpt/src/infer/cli_infer.py:220
↓ 1 callersFunctionload_code
(in_file)
speechgpt/utils/vocoder/generate_waveform_from_code.py:34
↓ 1 callersFunctionmain
(args)
speechgpt/utils/vocoder/generate_waveform_from_code.py:40
↓ 1 callersMethodmerge_duplicates
(cluster_ids)
speechgpt/utils/speech2unit/speech2unit.py:148
↓ 1 callersMethodpostprocess
( self, response: str, )
speechgpt/src/infer/cli_infer.py:126
↓ 1 callersMethodpreprocess
( self, raw_text: str, )
speechgpt/src/infer/cli_infer.py:110
↓ 1 callersMethodread_audio
(self, path)
speechgpt/utils/speech2unit/speech2unit.py:54
↓ 1 callersFunctionsafe_save_model_for_hf_trainer
Collects the state dict and dump to disk.
speechgpt/src/train/cm_sft.py:101
↓ 1 callersFunctionsafe_save_model_for_hf_trainer
Collects the state dict and dump to disk.
speechgpt/src/train/ma_pretrain.py:116
↓ 1 callersFunctiontrain
()
speechgpt/src/train/com_sft.py:150
↓ 1 callersFunctiontrain
()
speechgpt/src/train/cm_sft.py:111
↓ 1 callersFunctiontrain
()
speechgpt/src/train/ma_pretrain.py:125
Method__call__
(self, text, **kwargs)
speechgpt/utils/text2unit/text2unit.py:51
Method__call__
(self, x)
speechgpt/utils/speech2unit/speech2unit.py:102
Method__call__
(self, path, merged=True)
speechgpt/utils/speech2unit/speech2unit.py:162
Method__call__
(self, input)
speechgpt/src/infer/cli_infer.py:216
Method__init__
(self, verbose: bool = False)
speechgpt/utils/prompter.py:9
Method__init__
( self, checkpoint_dir="speechgpt/utils/text2unit", checkpoint_file="text2unit.pt",
speechgpt/utils/text2unit/text2unit.py:12
Method__init__
(self, ckpt_path, layer, max_chunk=1600000, fp16=False, sampling_rate=16000)
speechgpt/utils/speech2unit/speech2unit.py:35
Method__init__
Args: ckpt_dir(str): path to hubert model dir(e.g. hubert_base_ls960.pt) layer(int): feat from which layer of hubert
speechgpt/utils/speech2unit/speech2unit.py:122
Method__init__
( self, model_name_or_path: str, lora_weights: str=None, s2u_dir: str="speech
speechgpt/src/infer/cli_infer.py:51
Functiongenerate_and_tokenize_prompt
COM-style instructions
speechgpt/src/train/com_sft.py:264
Functiongenerate_and_tokenize_prompt
moss-style instructions
speechgpt/src/train/cm_sft.py:202
Functiongroup_texts
(examples)
speechgpt/src/train/ma_pretrain.py:277
Functionsafe_save_model_for_hf_trainer
Collects the state dict and dump to disk.
speechgpt/src/train/com_sft.py:140
Functionspeech_dialogue
(audio)
speechgpt/src/infer/web_infer.py:29
Functiontokenize_function
(examples)
speechgpt/src/train/ma_pretrain.py:232