hnluo
2023-08-07 607059f9d3433487c29a029e90296925476f9f17
Merge pull request #809 from alibaba-damo-academy/dev_lhn

fix modelscope inference bug
1个文件已修改
1 ■■■■ 已修改文件
egs/aishell/transformer/utils/prepare_checkpoint.py 1 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
egs/aishell/transformer/utils/prepare_checkpoint.py
@@ -16,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"))