to revert

huggingface · Jan 17, 2025 · fd11cd0 · fd11cd0
1 parent cf9a51d
commit fd11cd0
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/optimum/exporters/tflite/base.py b/optimum/exporters/tflite/base.py
@@ -27,6 +27,7 @@
 if is_tf_available():
     import tensorflow as tf
 
+
 from ..base import ExportersConfig
 
 
@@ -192,7 +193,16 @@ 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._config = config
+        self._normalized_config = self.NORMALIZED_CONFIG_CLASS(self._config)
+        self.mandatory_axes = ()
+        self.task = task
+        self._axes: Dict[str, int] = {}
+
+
+        # super().__init__(config=config, task=task, int_dtype="int64", float_dtype="fp32")
 
         # self.mandatory_axes = ()
         # self._axes: Dict[str, int] = {}

diff --git a/optimum/exporters/utils.py b/optimum/exporters/utils.py
@@ -545,7 +545,7 @@ def get_speecht5_models_for_export(
         use_past=use_past,
         use_past_in_inputs=False,  # Irrelevant here.
         behavior=config._behavior,  # Irrelevant here.
-        # preprocessors=config._preprocessors,
+        preprocessors=config._preprocessors,
         is_postnet_and_vocoder=True,
         legacy=config.legacy,
     )