From 18b1449d1ff06c469e54190508c4f6be05c73d85 Mon Sep 17 00:00:00 2001
From: 夜雨飘零 <yeyupiaoling@foxmail.com>
Date: 星期二, 05 十二月 2023 22:04:14 +0800
Subject: [PATCH] 分角色语音识别支持更多的模型
---
funasr/models/encoder/data2vec_encoder.py | 4 +---
1 files changed, 1 insertions(+), 3 deletions(-)
diff --git a/funasr/models/encoder/data2vec_encoder.py b/funasr/models/encoder/data2vec_encoder.py
index fd1796c..8885f02 100644
--- a/funasr/models/encoder/data2vec_encoder.py
+++ b/funasr/models/encoder/data2vec_encoder.py
@@ -10,7 +10,6 @@
import torch.distributed as dist
import torch.nn as nn
import torch.nn.functional as F
-from typeguard import check_argument_types
from funasr.models.encoder.abs_encoder import AbsEncoder
from funasr.modules.data2vec.data_utils import compute_mask_indices
@@ -97,7 +96,6 @@
# FP16 optimization
required_seq_len_multiple: int = 2,
):
- assert check_argument_types()
super().__init__()
# ConvFeatureExtractionModel
@@ -574,4 +572,4 @@
)
def output_size(self) -> int:
- return self.encoder_embed_dim
+ return self.encoder_embed_dim
\ No newline at end of file
--
Gitblit v1.9.1