From eff2570faf3dae7908db87edf4ef1a6ea88e5b33 Mon Sep 17 00:00:00 2001 From: cdevelop <cdevelop@qq.com> Date: 星期三, 15 十一月 2023 19:46:00 +0800 Subject: [PATCH] 解决windwos 加载lm模型失败 (#1093) --- funasr/models/frontend/abs_frontend.py | 17 +++++++++++++++++ 1 files changed, 17 insertions(+), 0 deletions(-) diff --git a/funasr/models/frontend/abs_frontend.py b/funasr/models/frontend/abs_frontend.py new file mode 100644 index 0000000..6049a01 --- /dev/null +++ b/funasr/models/frontend/abs_frontend.py @@ -0,0 +1,17 @@ +from abc import ABC +from abc import abstractmethod +from typing import Tuple + +import torch + + +class AbsFrontend(torch.nn.Module, ABC): + @abstractmethod + def output_size(self) -> int: + raise NotImplementedError + + @abstractmethod + def forward( + self, input: torch.Tensor, input_lengths: torch.Tensor + ) -> Tuple[torch.Tensor, torch.Tensor]: + raise NotImplementedError \ No newline at end of file -- Gitblit v1.9.1