From 7a40913af5003c0dd888bf4dcd168ecc0962aa3a Mon Sep 17 00:00:00 2001 From: Yingda Chen Date: Mon, 22 Aug 2022 17:35:37 +0800 Subject: [PATCH] [to #42322933] add constants --- modelscope/utils/constant.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/modelscope/utils/constant.py b/modelscope/utils/constant.py index 538aa3db..1d1c95ae 100644 --- a/modelscope/utils/constant.py +++ b/modelscope/utils/constant.py @@ -201,6 +201,8 @@ class ModelFile(object): TF_CKPT_PREFIX = 'ckpt-' TORCH_MODEL_FILE = 'pytorch_model.pt' TORCH_MODEL_BIN_FILE = 'pytorch_model.bin' + VOCAB_FILE = 'vocab.txt' + ONNX_MODEL_FILE = 'model.onnx' LABEL_MAPPING = 'label_mapping.json'