pull/3880/head
yinfan98 10 months ago
parent 1e74774409
commit 8026c6f2fc

@ -1 +1 @@
../../TTSCppFrontend/ ../../TTSCppFrontend

@ -1 +1 @@
../../../utils/ ../../../utils

@ -1 +1 @@
../../../tools/kaldi/egs/wsj/s5/steps/ ../../../tools/kaldi/egs/wsj/s5/steps

@ -1 +1 @@
../../../utils/ ../../../utils

@ -13,7 +13,3 @@
# limitations under the License. # limitations under the License.
import _locale import _locale
_locale._getdefaultlocale = (lambda *args: ['en_US', 'utf8']) _locale._getdefaultlocale = (lambda *args: ['en_US', 'utf8'])
__version__ = '0.0.0'
__commit__ = '9cf8c1985a98bb380c183116123672976bdfe5c9'

@ -17,11 +17,12 @@ from typing import Union
import numpy as np import numpy as np
import paddle import paddle
import paddle.nn.functional as F import paddle.nn.functional as F
import paddlespeech.s2t.modules.align as paddlespeech_nn
import soundfile import soundfile
import tqdm import tqdm
from paddle import nn from paddle import nn
from paddle.distribution import Categorical from paddle.distribution import Categorical
import paddlespeech.s2t.modules.align as paddlespeech_nn
from paddlespeech.s2t.models.whisper import utils from paddlespeech.s2t.models.whisper import utils
from paddlespeech.s2t.models.whisper.tokenizer import get_tokenizer from paddlespeech.s2t.models.whisper.tokenizer import get_tokenizer
from paddlespeech.s2t.models.whisper.tokenizer import LANGUAGES from paddlespeech.s2t.models.whisper.tokenizer import LANGUAGES
@ -475,8 +476,8 @@ def transcribe(
if dtype == np.float32: if dtype == np.float32:
decode_options["fp16"] = False decode_options["fp16"] = False
if decode_options.get( if decode_options.get("language") == 'None' or decode_options.get(
"language") == 'None' or decode_options.get("language", None) is None: "language", None) is None:
if not model.is_multilingual: if not model.is_multilingual:
decode_options["language"] = "en" decode_options["language"] = "en"
else: else:
@ -1206,8 +1207,9 @@ class DecodingTask:
DecodingResult( DecodingResult(
audio_features=features, audio_features=features,
language=language, language=language,
language_probs=probs) for features, language, probs in language_probs=probs)
zip(audio_features, languages, language_probs) for features, language, probs in zip(audio_features, languages,
language_probs)
] ]
# repeat the audio & text tensors by the group size, for beam search or best-of-n sampling # repeat the audio & text tensors by the group size, for beam search or best-of-n sampling

@ -1 +1 @@
../../../utils/ ../../../utils

@ -1 +1 @@
../../../../utils/ ../../../../utils
Loading…
Cancel
Save