From 64d703207304a062241f19eb19f6e3e7e8ab4399 Mon Sep 17 00:00:00 2001
From: 维石 <shixian.shi@alibaba-inc.com>
Date: 星期二, 28 五月 2024 19:09:03 +0800
Subject: [PATCH] update export
---
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