remove check_for_update

This commit is contained in:
XXXXRT666 2024-07-16 15:49:12 +08:00
parent bae36bb07a
commit bf292ccbe4
2 changed files with 6 additions and 1 deletions

View File

@ -4,7 +4,7 @@ import argparse
import os import os
import traceback import traceback
from tqdm import tqdm from tqdm import tqdm
from funasr.utils import version_checker
from funasr import AutoModel from funasr import AutoModel
path_asr = 'tools/asr/models/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch' path_asr = 'tools/asr/models/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch'
@ -14,6 +14,8 @@ path_asr = path_asr if os.path.exists(path_asr) else "iic/speech_paraformer-l
path_vad = path_vad if os.path.exists(path_vad) else "iic/speech_fsmn_vad_zh-cn-16k-common-pytorch" path_vad = path_vad if os.path.exists(path_vad) else "iic/speech_fsmn_vad_zh-cn-16k-common-pytorch"
path_punc = path_punc if os.path.exists(path_punc) else "iic/punc_ct-transformer_zh-cn-common-vocab272727-pytorch" path_punc = path_punc if os.path.exists(path_punc) else "iic/punc_ct-transformer_zh-cn-common-vocab272727-pytorch"
version_checker.check_for_update = lambda: None
model = AutoModel( model = AutoModel(
model = path_asr, model = path_asr,
model_revision = "v2.0.4", model_revision = "v2.0.4",

View File

@ -4,6 +4,7 @@ import re
import traceback import traceback
import torch import torch
from tqdm import tqdm from tqdm import tqdm
from funasr.utils import version_checker
from funasr import AutoModel from funasr import AutoModel
path_asr = "tools/asr/models/SenseVoiceSmall" path_asr = "tools/asr/models/SenseVoiceSmall"
@ -13,6 +14,8 @@ path_asr = path_asr if os.path.exists(path_asr) else "iic/SenseVoiceSmall"
path_vad = path_vad if os.path.exists(path_vad) else "iic/speech_fsmn_vad_zh-cn-16k-common-pytorch" path_vad = path_vad if os.path.exists(path_vad) else "iic/speech_fsmn_vad_zh-cn-16k-common-pytorch"
path_punc = path_punc if os.path.exists(path_punc) else "iic/punc_ct-transformer_zh-cn-common-vocab272727-pytorch" path_punc = path_punc if os.path.exists(path_punc) else "iic/punc_ct-transformer_zh-cn-common-vocab272727-pytorch"
version_checker.check_for_update = lambda: None
def execute_asr(input_folder, output_folder, language, device): def execute_asr(input_folder, output_folder, language, device):
try: try:
model = AutoModel(model=path_asr, model = AutoModel(model=path_asr,