Skip to content

Commit

Permalink
Export files required by OPT to load the tokenizer (#1571)
Browse files Browse the repository at this point in the history
* initial commit

* include Tuans proposal

* Update src/sparseml/transformers/export.py
  • Loading branch information
dbogunowicz committed May 22, 2023
1 parent 4fb5d27 commit 3dd1f8d
Showing 1 changed file with 8 additions and 3 deletions.
11 changes: 8 additions & 3 deletions src/sparseml/transformers/export.py
Original file line number Diff line number Diff line change
Expand Up @@ -91,13 +91,18 @@
__all__ = ["export_transformer_to_onnx", "load_task_model"]

MODEL_ONNX_NAME = "model.onnx"
EXTERNAL_ONNX_DATA_NAME = "model.data"
MANDATORY_DEPLOYMENT_FILES: List[str] = [
MANDATORY_DEPLOYMENT_FILES = [
MODEL_ONNX_NAME,
"tokenizer_config.json",
"config.json",
]
OPTIONAL_DEPLOYMENT_FILES: List[str] = [EXTERNAL_ONNX_DATA_NAME, "tokenizer.json"]
EXTERNAL_ONNX_DATA_NAME = ["model.data"]
OPT_TOKENIZER_FILES = ["special_tokens_map.json", "vocab.json", "merges.txt"]

OPTIONAL_DEPLOYMENT_FILES: List[str] = ["tokenizer.json"]
OPTIONAL_DEPLOYMENT_FILES.extend(EXTERNAL_ONNX_DATA_NAME)
OPTIONAL_DEPLOYMENT_FILES.extend(OPT_TOKENIZER_FILES)


_LOGGER = logging.getLogger(__name__)

Expand Down

0 comments on commit 3dd1f8d

Please sign in to comment.