o i@sFdZddlmZddlmZGdddeZdddd iZd d Zd S) zST Interface module.) ASRInterface)dynamic_importc@s*eZdZdZddgfddZdddZdS) STInterfacezST Interface for ESPnet model implementation. NOTE: This class is inherited from ASRInterface to enable joint translation and recognition when performing multi-task learning with the ASR task. NcCtd)aaRecognize x for evaluation. :param ndarray x: input acouctic feature (B, T, D) or (T, D) :param namespace trans_args: argment namespace contraining options :param list char_list: list of characters :param torch.nn.Module rnnlm: language model module :return: N-best decoding results :rtype: list z#translate method is not implementedNotImplementedError)selfx trans_args char_listrnnlmensemble_modelsrL/home/ubuntu/.local/lib/python3.10/site-packages/espnet/nets/st_interface.py translate zSTInterface.translatecCr)atBeam search implementation for batch. :param torch.Tensor x: encoder hidden state sequences (B, Tmax, Henc) :param namespace trans_args: argument namespace containing options :param list char_list: list of characters :param torch.nn.Module rnnlm: language model module :return: N-best decoding results :rtype: list z$Batch decoding is not supported yet.r)rr r r r rrrtranslate_batchrzSTInterface.translate_batch)NN)__name__ __module__ __qualname____doc__rrrrrrrs rpytorchz&espnet.nets.pytorch_backend.e2e_st:E2Ez2espnet.nets.pytorch_backend.e2e_st_transformer:E2E)rnn transformercCs0t|t|t}t|tsJ|d|S)zImport ST models dynamically. Args: module (str): module_name:class_name or alias in `predefined_st` backend (str): NN backend. e.g., pytorch, chainer Returns: type: ST class z does not implement STInterface)r predefined_stgetdict issubclassr)modulebackend model_classrrrdynamic_import_st4s r!N)respnet.nets.asr_interfacerespnet.utils.dynamic_importrrrr!rrrrs  "