Import the model.data from a single source variable in sparsezoo (#…

…1584)

* Update export.py

* quality

---------

Co-authored-by: Benjamin Fineran <bfineran@users.noreply.github.com>

src/sparseml/transformers/export.py

@@ -88,6 +88,7 @@
 from sparseml.pytorch.utils import export_onnx
 from sparseml.transformers.sparsification import Trainer
 from sparseml.transformers.utils import SparseAutoModel
+from sparsezoo.utils.onnx import EXTERNAL_ONNX_DATA_NAME
 
 
 __all__ = ["export_transformer_to_onnx", "load_task_model"]
@@ -98,11 +99,10 @@
     "tokenizer_config.json",
     "config.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.append(EXTERNAL_ONNX_DATA_NAME)
 OPTIONAL_DEPLOYMENT_FILES.extend(OPT_TOKENIZER_FILES)