Add separate V2 config

qubvel · qubvel · commit 2e0290b3cb60 · 2025-02-28T15:39:38.000Z
diff --git a/optimum/exporters/onnx/model_configs.py b/optimum/exporters/onnx/model_configs.py
@@ -2675,3 +2675,7 @@ def _create_dummy_input_generator_classes(self, **kwargs) -> List["DummyInputGen
             )
             kwargs["width"] = 320
         return super()._create_dummy_input_generator_classes(**kwargs)
+
+
+class RTDetrV2OnnxConfig(RTDetrOnnxConfig):
+    pass
diff --git a/optimum/exporters/tasks.py b/optimum/exporters/tasks.py
@@ -1110,7 +1110,7 @@ class TasksManager:
         ),
         "rt-detr-v2": supported_tasks_mapping(
             "object-detection",
-            onnx="RTDetrOnnxConfig",
+            onnx="RTDetrV2OnnxConfig",
         ),
         "sam": supported_tasks_mapping(
             "feature-extraction",

Original file line number	Diff line number	Diff line change
`@@ -2675,3 +2675,7 @@ def _create_dummy_input_generator_classes(self, **kwargs) -> List["DummyInputGen`
`2675`	`2675`	`)`
`2676`	`2676`	`kwargs["width"] = 320`
`2677`	`2677`	`return super()._create_dummy_input_generator_classes(**kwargs)`
	`2678`	`+`
	`2679`	`+`
	`2680`	`+class RTDetrV2OnnxConfig(RTDetrOnnxConfig):`
	`2681`	`+ pass`