From ac1d70ef0cd8fbbe54d637f68098851645f5eaee Mon Sep 17 00:00:00 2001
From: suluyana <110878454+suluyana@users.noreply.github.com>
Date: Wed, 18 Sep 2024 08:48:20 +0800
Subject: [PATCH] Fix problems with serializing audio output in serving (#993)

* fix audio out

* fix value in json output

* fix audio out

(cherry picked from commit d5c9c82340f39c0c63f32503725582a0959600aa)
---
 modelscope/utils/input_output.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/modelscope/utils/input_output.py b/modelscope/utils/input_output.py
index 37e875bc..50010baf 100644
--- a/modelscope/utils/input_output.py
+++ b/modelscope/utils/input_output.py
@@ -787,7 +787,12 @@ def pipeline_output_to_service_base64_output(task_name, pipeline_output):
         pipeline_output = pipeline_output[0]
     for key, value in pipeline_output.items():
         if key not in task_outputs:
-            json_serializable_output[key] = value
+            import torch
+            if isinstance(value, torch.Tensor):
+                v = np.array(value.cpu()).tolist()
+            else:
+                v = value
+            json_serializable_output[key] = v
             continue  # skip the output not defined.
         if key in [
                 OutputKeys.OUTPUT_IMG, OutputKeys.OUTPUT_IMGS,