From 0f1247d7a8d481a7b5825c903e819dc543224740 Mon Sep 17 00:00:00 2001
From: R1ckShi <shixian.shi@alibaba-inc.com>
Date: 星期二, 11 六月 2024 14:41:21 +0800
Subject: [PATCH] update scripts
---
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 8f1aa53..d205400 100644
--- a/funasr/utils/export_utils.py
+++ b/funasr/utils/export_utils.py
@@ -22,6 +22,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