diff --git a/optimum/exporters/onnx/model_configs.py b/optimum/exporters/onnx/model_configs.py index 776a1f8506..a8fa6c1e01 100644 --- a/optimum/exporters/onnx/model_configs.py +++ b/optimum/exporters/onnx/model_configs.py @@ -744,7 +744,7 @@ def generate(self, input_name: str, framework: str = "pt", int_dtype: str = "int class Dinov2OnnxConfig(ViTOnnxConfig): - DUMMY_INPUT_GENERATOR_CLASSES = (Dinov2DummyInputGenerator, ) + DUMMY_INPUT_GENERATOR_CLASSES = (Dinov2DummyInputGenerator,) class MobileViTOnnxConfig(ViTOnnxConfig):