support relative paths in sparseml.tarnsformers.export_onnx (#1699)

neuralmagic · Aug 10, 2023 · 45f051a · 45f051a
1 parent 8e4dc20
commit 45f051a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/sparseml/transformers/export.py b/src/sparseml/transformers/export.py
@@ -578,6 +578,9 @@ def export(
     data_args: Optional[str] = None,
     one_shot: Optional[str] = None,
 ):
+    if os.path.exists(model_path):
+        # expand to absolute path to support downstream logic
+        model_path = os.path.abspath(model_path)
     export_transformer_to_onnx(
         task=task,
         model_path=model_path,