blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-10-06 14:40:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

讲转写逻辑改为多进程利用多核cpu提高转写效率

Browse Source 
讲转写逻辑改为多进程利用多核cpu提高转写效率
This commit is contained in:
刘悦 2024-01-21 13:02:12 +08:00 committed by GitHub
parent 9886213592
commit ff0b37e7d9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tools/damo_asr/cmd-asr.py
View File

@ -30,7 +30,6 @@ inference_pipeline = pipeline(
punc_model='tools/damo_asr/models/punc_ct-transformer_zh-cn-common-vocab272727-pytorch', punc_model='tools/damo_asr/models/punc_ct-transformer_zh-cn-common-vocab272727-pytorch',
) )
def process_audio_file(dir,name,opt_name): def process_audio_file(dir,name,opt_name):
try: try:
@ -42,7 +41,6 @@ def process_audio_file(dir,name,opt_name):
except: except:
print(traceback.format_exc()) print(traceback.format_exc())
def run__process(): # 主进程 def run__process(): # 主进程
opt_dir="output/asr_opt" opt_dir="output/asr_opt"
@ -53,7 +51,6 @@ def run__process(): # 主进程
with multiprocessing.Pool(processes=processes) as pool: with multiprocessing.Pool(processes=processes) as pool:
pool.starmap(process_audio_file, [(dir, name ,opt_name) for name in os.listdir(dir)]) pool.starmap(process_audio_file, [(dir, name ,opt_name) for name in os.listdir(dir)])
if __name__ == '__main__': if __name__ == '__main__':
run__process() run__process()