Open loxs123 opened 3 weeks ago
目前解决方案
def worker_process():
wav_files = glob.glob('data/*/*/*.wav')
for wav_file in wav_files:
print('processing ',wav_file)
output_path =f'{wav_file[:-4]}.json'
start_time = time.time()
model= AutoModel(model="paraformer-zh", vad_model="fsmn-vad", punc_model="ct-punc", spk_model="cam++")
ans = model.generate(input=wav_file,batch_size_s=300,hotword='', )
os.system(f'rm "{wav_file}"')
end_time = time.time()
with open(output_path, 'w') as f:
json.dump(ans, f, ensure_ascii=False)
if __name__ == '__main__':
worker_process()
What is your question?
使用多次调用同一个模型报错,首次调用generate并不会报错,后2次或者3次就会报错
Code
What's your environment?
OS: CentOS Linux release 7.9.2009 FunASR Version: 1.0.27 ModelScope Version: 1.14.0 PyTorch Version: 2.1.2+cu118 How you installed funasr: pip Python version: 3.8.19 GPU: v100 CUDA/cuDNN version: 11.4 cpu