From cd001acf68f128245628667dc69c676104b2f438 Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期四, 21 三月 2024 12:52:02 +0800
Subject: [PATCH] Dev gzf train (#1524)
---
funasr/utils/misc.py | 1 +
funasr/models/paraformer/decoder.py | 4 ++--
2 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/funasr/models/paraformer/decoder.py b/funasr/models/paraformer/decoder.py
index f08e97b..b75d21d 100644
--- a/funasr/models/paraformer/decoder.py
+++ b/funasr/models/paraformer/decoder.py
@@ -351,9 +351,9 @@
hlens: torch.Tensor,
ys_in_pad: torch.Tensor,
ys_in_lens: torch.Tensor,
- return_hidden: bool = False,
- return_both: bool= False,
chunk_mask: torch.Tensor = None,
+ return_hidden: bool = False,
+ return_both: bool = False,
) -> Tuple[torch.Tensor, torch.Tensor]:
"""Forward decoder.
diff --git a/funasr/utils/misc.py b/funasr/utils/misc.py
index ef18a61..4b58d87 100644
--- a/funasr/utils/misc.py
+++ b/funasr/utils/misc.py
@@ -1,6 +1,7 @@
import os
import io
import shutil
+import logging
from collections import OrderedDict
import numpy as np
from omegaconf import DictConfig, OmegaConf
--
Gitblit v1.9.1