Skip to content

Commit fd11cd0

Browse files
committed
to revert
1 parent cf9a51d commit fd11cd0

File tree

2 files changed

+12
-2
lines changed

2 files changed

+12
-2
lines changed

optimum/exporters/tflite/base.py

+11-1
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
if is_tf_available():
2828
import tensorflow as tf
2929

30+
3031
from ..base import ExportersConfig
3132

3233

@@ -192,7 +193,16 @@ def __init__(
192193
point_batch_size: Optional[int] = None,
193194
nb_points_per_image: Optional[int] = None,
194195
):
195-
super().__init__(config=config, task=task, int_dtype="int64", float_dtype="fp32")
196+
197+
198+
self._config = config
199+
self._normalized_config = self.NORMALIZED_CONFIG_CLASS(self._config)
200+
self.mandatory_axes = ()
201+
self.task = task
202+
self._axes: Dict[str, int] = {}
203+
204+
205+
# super().__init__(config=config, task=task, int_dtype="int64", float_dtype="fp32")
196206

197207
# self.mandatory_axes = ()
198208
# self._axes: Dict[str, int] = {}

optimum/exporters/utils.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -545,7 +545,7 @@ def get_speecht5_models_for_export(
545545
use_past=use_past,
546546
use_past_in_inputs=False, # Irrelevant here.
547547
behavior=config._behavior, # Irrelevant here.
548-
# preprocessors=config._preprocessors,
548+
preprocessors=config._preprocessors,
549549
is_postnet_and_vocoder=True,
550550
legacy=config.legacy,
551551
)

0 commit comments

Comments
 (0)