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

fix modelscope inference bug
This commit is contained in:
hnluo 2023-08-07 11:14:13 +08:00 committed by GitHub
commit 607059f9d3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

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