Browse Source

[to #42322933]语音使用local import避免其他任务加载torchaudio

feat: local import torchaudio
        Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/9097950

    * feat: local import torchaudio
master
bin.xue 3 years ago
parent
commit
97a0087976
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      modelscope/preprocessors/audio.py

+ 2
- 1
modelscope/preprocessors/audio.py View File

@@ -5,7 +5,6 @@ from typing import Any, Dict
import numpy as np
import scipy.io.wavfile as wav
import torch
import torchaudio.compliance.kaldi as kaldi
from numpy.ctypeslib import ndpointer

from modelscope.utils.constant import Fields
@@ -123,6 +122,8 @@ class Feature:
if self.feat_type == 'raw':
return utt
elif self.feat_type == 'fbank':
# have to use local import before modelscope framework supoort lazy loading
import torchaudio.compliance.kaldi as kaldi
if len(utt.shape) == 1:
utt = utt.unsqueeze(0)
feat = kaldi.fbank(utt, **self.fbank_config)


Loading…
Cancel
Save