From 0a954637cb60cd8faaefcbe360f3f67af12672d2 Mon Sep 17 00:00:00 2001
From: speech_asr <wangjiaming.wjm@alibaba-inc.com>
Date: 星期二, 11 四月 2023 00:26:03 +0800
Subject: [PATCH] update
---
funasr/models/data2vec.py | 3 +--
1 files changed, 1 insertions(+), 2 deletions(-)
diff --git a/funasr/models/data2vec.py b/funasr/models/data2vec.py
index 2d4711f..380c137 100644
--- a/funasr/models/data2vec.py
+++ b/funasr/models/data2vec.py
@@ -12,7 +12,6 @@
import torch
from typeguard import check_argument_types
-from funasr.layers.abs_normalize import AbsNormalize
from funasr.models.preencoder.abs_preencoder import AbsPreEncoder
from funasr.torch_utils.device_funcs import force_gatherable
from funasr.models.base_model import FunASRModel
@@ -33,7 +32,7 @@
self,
frontend: Optional[torch.nn.Module],
specaug: Optional[torch.nn.Module],
- normalize: Optional[AbsNormalize],
+ normalize: Optional[torch.nn.Module],
preencoder: Optional[AbsPreEncoder],
encoder: torch.nn.Module,
):
--
Gitblit v1.9.1