From 5ee2f382b32c5c0beaa1c133b466a100c6fb7ebc Mon Sep 17 00:00:00 2001
From: passerbya <hanghang3103@163.com>
Date: 星期四, 20 三月 2025 23:01:05 +0800
Subject: [PATCH] FIX 'NoneType' object has no attribute 'isalpha' (#2440)
---
funasr/models/contextual_paraformer/export_meta.py | 17 ++++++++++++++++-
1 files changed, 16 insertions(+), 1 deletions(-)
diff --git a/funasr/models/contextual_paraformer/export_meta.py b/funasr/models/contextual_paraformer/export_meta.py
index 7543789..9d3a63b 100644
--- a/funasr/models/contextual_paraformer/export_meta.py
+++ b/funasr/models/contextual_paraformer/export_meta.py
@@ -16,6 +16,21 @@
self.embedding = model.bias_embed
model.bias_encoder.batch_first = False
self.bias_encoder = model.bias_encoder
+
+ def export_dummy_inputs(self):
+ hotword = torch.tensor(
+ [
+ [10, 11, 12, 13, 14, 10, 11, 12, 13, 14],
+ [100, 101, 0, 0, 0, 0, 0, 0, 0, 0],
+ [1, 0, 0, 0, 0, 0, 0, 0, 0, 0],
+ [10, 11, 12, 13, 14, 10, 11, 12, 13, 14],
+ [100, 101, 0, 0, 0, 0, 0, 0, 0, 0],
+ [1, 0, 0, 0, 0, 0, 0, 0, 0, 0],
+ ],
+ dtype=torch.int32,
+ )
+ # hotword_length = torch.tensor([10, 2, 1], dtype=torch.int32)
+ return (hotword)
def export_rebuild_model(model, **kwargs):
@@ -61,7 +76,7 @@
)
embedder_model.export_name = "model_eb"
- backbone_model.export_name = "model_bb"
+ backbone_model.export_name = "model"
return backbone_model, embedder_model
--
Gitblit v1.9.1