From 23eed38e9fbe547bab63a99a560c0ad33a5f13b8 Mon Sep 17 00:00:00 2001
From: 维石 <shixian.shi@alibaba-inc.com>
Date: 星期四, 30 五月 2024 14:22:32 +0800
Subject: [PATCH] update export name
---
funasr/utils/export_utils.py | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/funasr/utils/export_utils.py b/funasr/utils/export_utils.py
index 7d6606b..1deeaa7 100644
--- a/funasr/utils/export_utils.py
+++ b/funasr/utils/export_utils.py
@@ -78,7 +78,7 @@
)
-def _torchscripts(model, path, device='cpu'):
+def _torchscripts(model, path, device='cuda'):
dummy_input = model.export_dummy_inputs()
if device == 'cuda':
--
Gitblit v1.9.1