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/auto/auto_model.py | 8 +-------
1 files changed, 1 insertions(+), 7 deletions(-)
diff --git a/funasr/auto/auto_model.py b/funasr/auto/auto_model.py
index 97eb325..faa5bd9 100644
--- a/funasr/auto/auto_model.py
+++ b/funasr/auto/auto_model.py
@@ -580,12 +580,6 @@
)
with torch.no_grad():
-
- if type == "onnx":
- export_dir = export_utils.export_onnx(model=model, data_in=data_list, **kwargs)
- else:
- export_dir = export_utils.export_torchscripts(
- model=model, data_in=data_list, **kwargs
- )
+ export_dir = export_utils.export(model=model, data_in=data_list, **kwargs)
return export_dir
--
Gitblit v1.9.1