sensevoice

This commit is contained in:
游雁 2024-07-17 09:31:16 +08:00
parent 586287df30
commit b5c8b0e415
2 changed files with 4 additions and 0 deletions

View File

@ -110,6 +110,8 @@ model_dir = "iic/SenseVoiceSmall"
model = AutoModel(
model=model_dir,
trust_remote_code=True,
remote_code="./model.py",
vad_model="fsmn-vad",
vad_kwargs={"max_single_segment_time": 30000},
device="cpu",

View File

@ -12,6 +12,8 @@ model_dir = "iic/SenseVoiceSmall"
model = AutoModel(
model=model_dir,
trust_remote_code=True,
remote_code="./model.py",
vad_model="fsmn-vad",
vad_kwargs={"max_single_segment_time": 30000},
device="cuda:0",