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'