Skip to content

Commit

Permalink
Update HFOnnx pipeline to default to opset 14, closes #719
Browse files Browse the repository at this point in the history
  • Loading branch information
davidmezzetti committed May 18, 2024
1 parent 8bd4d78 commit 03057f2
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/python/txtai/pipeline/train/hfonnx.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ class HFOnnx(Tensors):
Exports a Hugging Face Transformer model to ONNX.
"""

def __call__(self, path, task="default", output=None, quantize=False, opset=12):
def __call__(self, path, task="default", output=None, quantize=False, opset=14):
"""
Exports a Hugging Face Transformer model to ONNX.
Expand All @@ -38,7 +38,7 @@ def __call__(self, path, task="default", output=None, quantize=False, opset=12):
task: optional model task or category, determines the model type and outputs, defaults to export hidden state
output: optional output model path, defaults to return byte array if None
quantize: if model should be quantized (requires onnx to be installed), defaults to False
opset: onnx opset, defaults to 12
opset: onnx opset, defaults to 14
Returns:
path to model output or model as bytes depending on output parameter
Expand Down

0 comments on commit 03057f2

Please sign in to comment.