From e04489ce4c0fd0095d0c79ef8f504f425e0435a8 Mon Sep 17 00:00:00 2001 From: Shi Xian <40013335+R1ckShi@users.noreply.github.com> Date: 星期三, 13 三月 2024 16:34:42 +0800 Subject: [PATCH] contextual&seaco ONNX export (#1481) --- funasr/models/transducer/beam_search_transducer.py | 10 ++++++---- 1 files changed, 6 insertions(+), 4 deletions(-) diff --git a/funasr/models/transducer/beam_search_transducer.py b/funasr/models/transducer/beam_search_transducer.py index 04b26b3..f599615 100644 --- a/funasr/models/transducer/beam_search_transducer.py +++ b/funasr/models/transducer/beam_search_transducer.py @@ -1,10 +1,12 @@ -"""Search algorithms for Transducer models.""" +#!/usr/bin/env python3 +# -*- encoding: utf-8 -*- +# Copyright FunASR (https://github.com/alibaba-damo-academy/FunASR). All Rights Reserved. +# MIT License (https://opensource.org/licenses/MIT) +import torch +import numpy as np from dataclasses import dataclass from typing import Any, Dict, List, Optional, Tuple, Union - -import numpy as np -import torch from funasr.models.transducer.joint_network import JointNetwork -- Gitblit v1.9.1