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 numpy as np
import scipy.io.wavfile as wav import scipy.io.wavfile as wav
import torch import torch
import torchaudio.compliance.kaldi as kaldi
from numpy.ctypeslib import ndpointer from numpy.ctypeslib import ndpointer


from modelscope.utils.constant import Fields from modelscope.utils.constant import Fields
@@ -123,6 +122,8 @@ class Feature:
if self.feat_type == 'raw': if self.feat_type == 'raw':
return utt return utt
elif self.feat_type == 'fbank': 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: if len(utt.shape) == 1:
utt = utt.unsqueeze(0) utt = utt.unsqueeze(0)
feat = kaldi.fbank(utt, **self.fbank_config) feat = kaldi.fbank(utt, **self.fbank_config)


Loading…
Cancel
Save