From 5052ca8bf03c178d9ae73cb68785cd4afb0144d2 Mon Sep 17 00:00:00 2001
From: 辰冢 <49506152+BruceLee569@users.noreply.github.com>
Date: 星期三, 12 二月 2025 16:13:08 +0800
Subject: [PATCH] Hotwords file needs to specify default utf-8 encoding. (#2379)
---
funasr/models/bicif_paraformer/export_meta.py | 42 +++++++++++++++++++-----------------------
1 files changed, 19 insertions(+), 23 deletions(-)
diff --git a/funasr/models/bicif_paraformer/export_meta.py b/funasr/models/bicif_paraformer/export_meta.py
index 7ae800e..75171f4 100644
--- a/funasr/models/bicif_paraformer/export_meta.py
+++ b/funasr/models/bicif_paraformer/export_meta.py
@@ -8,6 +8,7 @@
from funasr.register import tables
+
def export_rebuild_model(model, **kwargs):
is_onnx = kwargs.get("type", "onnx") == "onnx"
encoder_class = tables.encoder_classes.get(kwargs["encoder"] + "Export")
@@ -21,16 +22,18 @@
from funasr.utils.torch_function import sequence_mask
- model.make_pad_mask = sequence_mask(kwargs['max_seq_len'], flip=False)
+ model.make_pad_mask = sequence_mask(kwargs["max_seq_len"], flip=False)
model.forward = types.MethodType(export_forward, model)
model.export_dummy_inputs = types.MethodType(export_dummy_inputs, model)
model.export_input_names = types.MethodType(export_input_names, model)
model.export_output_names = types.MethodType(export_output_names, model)
model.export_dynamic_axes = types.MethodType(export_dynamic_axes, model)
- model.export_name = types.MethodType(export_name, model)
+
+ model.export_name = "model"
return model
+
def export_forward(
self,
@@ -53,39 +56,32 @@
return decoder_out, pre_token_length, us_alphas, us_cif_peak
+
def export_dummy_inputs(self):
speech = torch.randn(2, 30, 560)
speech_lengths = torch.tensor([6, 30], dtype=torch.int32)
return (speech, speech_lengths)
+
def export_input_names(self):
- return ['speech', 'speech_lengths']
+ return ["speech", "speech_lengths"]
+
def export_output_names(self):
- return ['logits', 'token_num', 'us_alphas', 'us_cif_peak']
+ return ["logits", "token_num", "us_alphas", "us_cif_peak"]
+
def export_dynamic_axes(self):
return {
- 'speech': {
- 0: 'batch_size',
- 1: 'feats_length'
+ "speech": {0: "batch_size", 1: "feats_length"},
+ "speech_lengths": {
+ 0: "batch_size",
},
- 'speech_lengths': {
- 0: 'batch_size',
- },
- 'logits': {
- 0: 'batch_size',
- 1: 'logits_length'
- },
- 'us_alphas': {
- 0: 'batch_size',
- 1: 'alphas_length'
- },
- 'us_cif_peak': {
- 0: 'batch_size',
- 1: 'alphas_length'
- },
+ "logits": {0: "batch_size", 1: "logits_length"},
+ "us_alphas": {0: "batch_size", 1: "alphas_length"},
+ "us_cif_peak": {0: "batch_size", 1: "alphas_length"},
}
+
def export_name(self):
- return "model.onnx"
\ No newline at end of file
+ return "model.onnx"
--
Gitblit v1.9.1