setup jamo

游雁

2023-11-24 b5d3df75cf6462aa3bf42fd3c86fa2aa7f1c8a15

 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"))