From 62ae9989266eb0d793d050ab50ea9c70528fddc7 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期五, 14 六月 2024 14:44:11 +0800
Subject: [PATCH] decoding
---
funasr/models/llm_asr/model.py | 6 ++++--
1 files changed, 4 insertions(+), 2 deletions(-)
diff --git a/funasr/models/llm_asr/model.py b/funasr/models/llm_asr/model.py
index 84d7d33..1151269 100644
--- a/funasr/models/llm_asr/model.py
+++ b/funasr/models/llm_asr/model.py
@@ -398,7 +398,9 @@
# frontend = model.kwargs.get("frontend")
audio_encoder_output_size = model.model.encoder_output_size
- audio_encoder = model.model.model.encoder
+ audio_encoder = (
+ model.model.model.encoder if hasattr(model.model, "model") else model.model.encoder
+ )
# self.frontend = frontend
@@ -814,7 +816,7 @@
@tables.register("model_classes", "LLMASR3")
-class LLMASR3(nn.Module):
+class LLMASR3(LLMASR2):
""" """
def __init__(
--
Gitblit v1.9.1