From 46568227c9ff0a1f6ebf9e398dfdb6ad51cce4c5 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期三, 05 六月 2024 10:24:01 +0800
Subject: [PATCH] auto frontend
---
funasr/models/llm_asr/adaptor.py | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/funasr/models/llm_asr/adaptor.py b/funasr/models/llm_asr/adaptor.py
index 75494de..1928b52 100644
--- a/funasr/models/llm_asr/adaptor.py
+++ b/funasr/models/llm_asr/adaptor.py
@@ -84,7 +84,7 @@
self.blocks = nn.ModuleList(
[
EncoderLayer(
- output_size,
+ llm_dim,
MultiHeadedAttention(
kwargs.get("attention_heads", 8),
llm_dim,
--
Gitblit v1.9.1