From ede2b4223d50108e4656bfd6de2133a71d920cb2 Mon Sep 17 00:00:00 2001
From: Ella Charlaix <ella@huggingface.co>
Date: Mon, 2 Oct 2023 18:36:47 +0200
Subject: [PATCH] format

---
 optimum/exporters/onnx/model_configs.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/optimum/exporters/onnx/model_configs.py b/optimum/exporters/onnx/model_configs.py
index 374c35feb7e..77f958de7d9 100644
--- a/optimum/exporters/onnx/model_configs.py
+++ b/optimum/exporters/onnx/model_configs.py
@@ -226,7 +226,6 @@ class LlamaOnnxConfig(TextDecoderWithPositionIdsOnnxConfig):
 class MistralOnnxConfig(TextDecoderWithPositionIdsOnnxConfig):
     # The ONNX export of this architecture needs the Trilu operator support, available since opset 14
     DEFAULT_ONNX_OPSET = 14
-    # NORMALIZED_CONFIG_CLASS = NormalizedTextConfig.with_args( num_attention_heads="num_key_value_heads", hidden_size="hidden_size", num_layers="num_hidden_layers")
     DUMMY_INPUT_GENERATOR_CLASSES = (
         MistralDummyPastKeyValuesGenerator,
     ) + TextDecoderOnnxConfig.DUMMY_INPUT_GENERATOR_CLASSES