From fa9a6cdb1eade68c258eed7297f5a8a8a5329ac6 Mon Sep 17 00:00:00 2001
From: Flute <41096447+fluteink@users.noreply.github.com>
Date: 星期三, 01 十月 2025 14:44:28 +0800
Subject: [PATCH] 更新文档和运行脚本，修复文档拼写错误 (#2688)
---
funasr/models/emotion2vec/export_meta.py | 13 +++----------
1 files changed, 3 insertions(+), 10 deletions(-)
diff --git a/funasr/models/emotion2vec/export_meta.py b/funasr/models/emotion2vec/export_meta.py
index 2954e5f..f0c5f43 100644
--- a/funasr/models/emotion2vec/export_meta.py
+++ b/funasr/models/emotion2vec/export_meta.py
@@ -21,13 +21,10 @@
model.export_dynamic_axes = types.MethodType(export_dynamic_axes, model)
model.export_name = types.MethodType(export_name, model)
- model.export_name = "emotion2vec"
return model
-def export_forward(
- self, x: torch.Tensor
-):
+def export_forward(self, x: torch.Tensor):
with torch.no_grad():
if self.cfg.normalize:
mean = torch.mean(x, dim=1, keepdim=True)
@@ -38,13 +35,9 @@
# Call the original forward directly just like extract_features
# Cannot directly use self.extract_features since it is being replaced by export_forward
res = self._original_forward(
- source=x,
- padding_mask=None,
- mask=False,
- features_only=True,
- remove_extra_tokens=True
+ source=x, padding_mask=None, mask=False, features_only=True, remove_extra_tokens=True
)
-
+
x = res["x"]
return x
--
Gitblit v1.9.1