From d43d0853dcf3a1db04302c7b527e92ace3ccfb55 Mon Sep 17 00:00:00 2001
From: AldarisX <aldaris@axnet.icu>
Date: 星期一, 07 四月 2025 21:20:31 +0800
Subject: [PATCH] add intel xpu support (#2468)
---
funasr/models/emotion2vec/model.py | 15 ++++++++++++++-
1 files changed, 14 insertions(+), 1 deletions(-)
diff --git a/funasr/models/emotion2vec/model.py b/funasr/models/emotion2vec/model.py
index 48b8716..75ede4c 100644
--- a/funasr/models/emotion2vec/model.py
+++ b/funasr/models/emotion2vec/model.py
@@ -249,12 +249,25 @@
if self.proj:
x = x.mean(dim=1)
x = self.proj(x)
+ for idx, lab in enumerate(labels):
+ x[:,idx] = -np.inf if lab.startswith("unuse") else x[:,idx]
x = torch.softmax(x, dim=-1)
scores = x[0].tolist()
- result_i = {"key": key[i], "labels": labels, "scores": scores}
+ select_label = [lb for lb in labels if not lb.startswith("unuse")]
+ select_score = [scores[idx] for idx, lb in enumerate(labels) if not lb.startswith("unuse")]
+
+ # result_i = {"key": key[i], "labels": labels, "scores": scores}
+ result_i = {"key": key[i], "labels": select_label, "scores": select_score}
+
if extract_embedding:
result_i["feats"] = feats
results.append(result_i)
return results, meta_data
+
+ def export(self, **kwargs):
+ from .export_meta import export_rebuild_model
+
+ models = export_rebuild_model(model=self, **kwargs)
+ return models
--
Gitblit v1.9.1