Add large test for RT-DETR

huggingface · Jun 28, 2024 · b3f4259 · b3f4259
1 parent 16d03eb
commit b3f4259
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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",