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 @@ -18,3 +16,4 @@ 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"))
