Updated docstring and removed redundant code

huggingface · Nov 22, 2022 · 28dc7d0 · 28dc7d0
1 parent 96cdd90
commit 28dc7d0
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 6 deletions.
diff --git a/optimum/exporters/onnx/base.py b/optimum/exporters/onnx/base.py
@@ -334,7 +334,7 @@ def get_encoder_onnx_config(self, config: "PretrainedConfig"):
 
     def get_decoder_onnx_config(self, config: "PretrainedConfig", task: str = "default", use_past: bool = False):
         """
-        Returns ONNX decoder config for `Seq2Seq` model. Implement the method to export the encoder
+        Returns ONNX decoder config for `Seq2Seq` models. Implement the method to export the decoder
         of the model separately.
 
         Args:

diff --git a/optimum/utils/input_generators.py b/optimum/utils/input_generators.py
@@ -27,13 +27,9 @@
 if is_torch_available():
     import torch
 
-    torch.manual_seed(0)
-
 if is_tf_available():
     import tensorflow as tf
 
-    tf.random.set_seed(0)
-
 
 def check_framework_is_available(func):
     @functools.wraps(func)

diff --git a/tests/exporters/test_onnx_export.py b/tests/exporters/test_onnx_export.py
@@ -308,7 +308,7 @@ def _onnx_export_for_encoder_decoder_models_for_ort(
             decoder_onnx_config_with_past = onnx_config.get_decoder_onnx_config(
                 decoder_model.config, task.replace("-with-past", ""), use_past=True
             )
-            models_for_export.extend([(model, decoder_onnx_config_with_past)])
+            models_for_export.append((model, decoder_onnx_config_with_past))
 
         for model_to_export, model_onnx_config in models_for_export:
             with NamedTemporaryFile("w") as output: