From a0c58c8fc6fd68055897be95d11be35506581a91 Mon Sep 17 00:00:00 2001
From: R1ckShi <shixian.shi@alibaba-inc.com>
Date: 星期二, 11 六月 2024 14:41:24 +0800
Subject: [PATCH] Merge branch 'dev_libt' of https://github.com/alibaba-damo-academy/FunASR into dev_libt
---
funasr/utils/export_utils.py | 1 +
1 files changed, 1 insertions(+), 0 deletions(-)
diff --git a/funasr/utils/export_utils.py b/funasr/utils/export_utils.py
index 6583325..32a88ab 100644
--- a/funasr/utils/export_utils.py
+++ b/funasr/utils/export_utils.py
@@ -28,6 +28,7 @@
)
elif type == 'torchscripts':
device = 'cuda' if torch.cuda.is_available() else 'cpu'
+ print("Exporting torchscripts on device {}".format(device))
_torchscripts(
m,
path=export_dir,
--
Gitblit v1.9.1