format

huggingface · Oct 2, 2023 · ede2b42 · ede2b42
1 parent 6c4dc3d
commit ede2b42
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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