莫拉古
2024-11-29 ae49b2a8e1bc676e6014d8a12ebeec947b655e3e
examples/industrial_data_pretraining/sense_voice/demo.py
@@ -1,8 +1,7 @@
#!/usr/bin/env python3
# -*- encoding: utf-8 -*-
# Copyright FunASR (https://github.com/alibaba-damo-academy/FunASR). All Rights Reserved.
# Copyright FunASR (https://github.com/FunAudioLLM/SenseVoice). All Rights Reserved.
#  MIT License  (https://opensource.org/licenses/MIT)
from funasr import AutoModel
from funasr.utils.postprocess_utils import rich_transcription_postprocess
@@ -14,7 +13,7 @@
    model=model_dir,
    vad_model="fsmn-vad",
    vad_kwargs={"max_single_segment_time": 30000},
    device="cpu",
    device="cuda:0",
)
# en