From b3f425992ea87aeb05879734f217e6edce8fc569 Mon Sep 17 00:00:00 2001 From: Joshua Lochner Date: Fri, 28 Jun 2024 15:43:00 +0000 Subject: [PATCH] Add large test for RT-DETR --- tests/exporters/exporters_utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/exporters/exporters_utils.py b/tests/exporters/exporters_utils.py index 0c52754ff60..7f2662c0c2e 100644 --- a/tests/exporters/exporters_utils.py +++ b/tests/exporters/exporters_utils.py @@ -260,6 +260,7 @@ "resnet": "microsoft/resnet-50", "roberta": "roberta-base", "roformer": "junnyu/roformer_chinese_base", + "rt-detr": "PekingU/rtdetr_r50vd", "sam": "facebook/sam-vit-base", "segformer": "nvidia/segformer-b0-finetuned-ade-512-512", "splinter": "hf-internal-testing/tiny-random-SplinterModel",