Skip to content

Commit

Permalink
make style
Browse files Browse the repository at this point in the history
  • Loading branch information
tedasdf committed Nov 6, 2024
1 parent c7b7d21 commit 5aae9cb
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 2 deletions.
3 changes: 2 additions & 1 deletion optimum/exporters/onnx/model_configs.py
Original file line number Diff line number Diff line change
Expand Up @@ -117,6 +117,7 @@ def inputs(self) -> Dict[str, Dict[int, str]]:
"token_type_ids": dynamic_axis,
}


class VisualBertOnnxConfig(TextAndVisionOnnxConfig):
DEFAULT_ONNX_OPSET = 11

Expand All @@ -127,7 +128,7 @@ def inputs(self) -> Dict[str, Dict[int, str]]:
"attention_mask": {0: "batch_size", 1: "sequence_length"},
"pixel_values": {0: "batch_size", 1: "num_channels", 2: "height", 3: "width"},
}

@property
def outputs(self) -> Dict[str, Dict[int, str]]:
return {
Expand Down
2 changes: 1 addition & 1 deletion tests/exporters/exporters_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -197,7 +197,7 @@
"document-question-answering-with-past",
],
},
"visualbert": "hf-internal-testing/tiny-random-VisualBertModel",
"visualbert": "hf-internal-testing/tiny-random-VisualBertModel",
}


Expand Down

0 comments on commit 5aae9cb

Please sign in to comment.