ソースを参照

Merge pull request #809 from alibaba-damo-academy/dev_lhn

fix modelscope inference bug
hnluo 2 年 前
コミット
607059f9d3
1 ファイル変更1 行追加0 行削除
  1. 1 0
      egs/aishell/transformer/utils/prepare_checkpoint.py

+ 1 - 0
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"))