From c4ac64fd5d24bb3fc8ccc441d36a07c83c8b9015 Mon Sep 17 00:00:00 2001
From: Yu Cao <monstercy@hotmail.com>
Date: 星期三, 01 十月 2025 14:46:21 +0800
Subject: [PATCH] fix "can not find model issue when running libtorch runtime" (#2504)

---
 funasr/models/paraformer/export_meta.py |    2 ++
 1 files changed, 2 insertions(+), 0 deletions(-)

diff --git a/funasr/models/paraformer/export_meta.py b/funasr/models/paraformer/export_meta.py
index 5c1b6c0..8e086a2 100644
--- a/funasr/models/paraformer/export_meta.py
+++ b/funasr/models/paraformer/export_meta.py
@@ -31,6 +31,7 @@
     model.export_dynamic_axes = types.MethodType(export_dynamic_axes, model)
     model.export_name = types.MethodType(export_name, model)
 
+    model.export_name = "model"
     return model
 
 
@@ -76,6 +77,7 @@
             0: "batch_size",
         },
         "logits": {0: "batch_size", 1: "logits_length"},
+        "token_num": {0: "batch_size"}
     }
 
 

--
Gitblit v1.9.1