From cf36ce977c0b8bd54c906e63ce31931ac060178f Mon Sep 17 00:00:00 2001 From: anyvoice <130631963+anyvoiceai@users.noreply.github.com> Date: 星期日, 02 七月 2023 09:12:18 +0800 Subject: [PATCH] Creates tensor in target device to avoid high CPU occupation. (#695) --- egs/aishell/transformer/utils/prepare_checkpoint.py | 2 -- 1 files changed, 0 insertions(+), 2 deletions(-) diff --git a/egs/aishell/transformer/utils/prepare_checkpoint.py b/egs/aishell/transformer/utils/prepare_checkpoint.py index 1c732e4..4d5e8a0 100644 --- a/egs/aishell/transformer/utils/prepare_checkpoint.py +++ b/egs/aishell/transformer/utils/prepare_checkpoint.py @@ -1,8 +1,6 @@ import os import shutil -from modelscope.pipelines import pipeline -from modelscope.utils.constant import Tasks from modelscope.hub.snapshot_download import snapshot_download -- Gitblit v1.9.1