| | |
|---|
| | | from packaging.version import parse as V |
|---|
| | | from typeguard import check_argument_types, check_return_type |
|---|
| | | |
|---|
| | | from funasr.models_transducer.beam_search_transducer import ( |
|---|
| | | from funasr.modules.beam_search.beam_search_transducer import ( |
|---|
| | | BeamSearchTransducer, |
|---|
| | | Hypothesis, |
|---|
| | | ) |
|---|
| | | from funasr.models_transducer.utils import TooShortUttError |
|---|
| | | from funasr.modules.nets_utils import TooShortUttError |
|---|
| | | from funasr.fileio.datadir_writer import DatadirWriter |
|---|
| | | from funasr.tasks.asr_transducer import ASRTransducerTask |
|---|
| | | from funasr.tasks.lm import LMTask |
|---|
| | |
|---|
| | | |
|---|
| | | feats = speech.unsqueeze(0).to(getattr(torch, self.dtype)) |
|---|
| | | feats_lengths = feats.new_full([1], dtype=torch.long, fill_value=feats.size(1)) |
|---|
| | | |
|---|
| | | if self.asr_model.normalize is not None: |
|---|
| | | feats, feats_lengths = self.asr_model.normalize(feats, feats_lengths) |
|---|
| | | |
|---|
| | | feats = to_device(feats, device=self.device) |
|---|
| | | feats_lengths = to_device(feats_lengths, device=self.device) |
|---|
