From 57e023e5cfa522007e24d87c8e3d82bf7c4a19cd Mon Sep 17 00:00:00 2001
From: Shi Xian <40013335+R1ckShi@users.noreply.github.com>
Date: 星期二, 18 二月 2025 15:15:40 +0800
Subject: [PATCH] Merge pull request #2382 from msgk239/dev_clean
---
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