From 6849c028be5ae4df417392300242ab7d544d2616 Mon Sep 17 00:00:00 2001 From: "haoneng.lhn" Date: Mon, 7 Aug 2023 11:12:26 +0800 Subject: [PATCH] fix modelscope inference bug --- egs/aishell/transformer/utils/prepare_checkpoint.py | 1 + 1 file changed, 1 insertion(+) diff --git a/egs/aishell/transformer/utils/prepare_checkpoint.py b/egs/aishell/transformer/utils/prepare_checkpoint.py index 4d5e8a085..116fa078b 100644 --- a/egs/aishell/transformer/utils/prepare_checkpoint.py +++ b/egs/aishell/transformer/utils/prepare_checkpoint.py @@ -16,3 +16,4 @@ if __name__ == '__main__': except BaseException: raise BaseException(f"Please download pretrain model from ModelScope firstly.") shutil.copy(os.path.join(checkpoint_dir, checkpoint_name), os.path.join(pretrained_model_path, "model.pb")) + shutil.copy(os.path.join(checkpoint_dir, "config.yaml"), os.path.join(pretrained_model_path, "config.yaml"))