Skip to content

Commit

Permalink
Update optimum/exporters/onnx/model_configs.py
Browse files Browse the repository at this point in the history
Co-authored-by: Ella Charlaix <[email protected]>
  • Loading branch information
IlyasMoutawwakil and echarlaix authored Sep 13, 2024
1 parent 002ed68 commit 70e4577
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion optimum/exporters/onnx/model_configs.py
Original file line number Diff line number Diff line change
Expand Up @@ -1112,7 +1112,7 @@ def ordered_inputs(self, model) -> Dict[str, Dict[int, str]]:

class VaeEncoderOnnxConfig(VisionOnnxConfig):
ATOL_FOR_VALIDATION = 1e-4
# The ONNX export of a VaeEncoder architecture, an other Stable Diffusion component, needs the Trilu
# The ONNX export of a CLIPText architecture, an other Stable Diffusion component, needs the Trilu
# operator support, available since opset 14
DEFAULT_ONNX_OPSET = 14

Expand Down

0 comments on commit 70e4577

Please sign in to comment.