Skip to content

Commit 07a1368

Browse files
committed
Add rt-detr-v2
1 parent 9a4a4e3 commit 07a1368

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

optimum/exporters/tasks.py

+4
Original file line numberDiff line numberDiff line change
@@ -1108,6 +1108,10 @@ class TasksManager:
11081108
"object-detection",
11091109
onnx="RTDetrOnnxConfig",
11101110
),
1111+
"rt-detr-v2": supported_tasks_mapping(
1112+
"object-detection",
1113+
onnx="RTDetrOnnxConfig",
1114+
),
11111115
"sam": supported_tasks_mapping(
11121116
"feature-extraction",
11131117
onnx="SamOnnxConfig",

0 commit comments

Comments
 (0)