Skip to content

Commit

Permalink
task attribute already set
Browse files Browse the repository at this point in the history
  • Loading branch information
echarlaix committed Jan 31, 2025
1 parent 8a936d2 commit bc423a5
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions optimum/exporters/tflite/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -169,10 +169,10 @@ def __init__(
point_batch_size: Optional[int] = None,
nb_points_per_image: Optional[int] = None,
):
super().__init__(config=config, task=task, int_dtype="int64", float_dtype="fp32")
self.mandatory_axes = ()
self._axes: Dict[str, int] = {}
self.task = task

super().__init__(config=config, task=task, int_dtype="int64", float_dtype="fp32")

# To avoid using **kwargs.
axes_values = {
Expand Down

0 comments on commit bc423a5

Please sign in to comment.