fix expected configs model type

huggingface · Dec 17, 2024 · f25f227 · f25f227
1 parent 1f71fae
commit f25f227
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/exporters/exporters_utils.py b/tests/exporters/exporters_utils.py
@@ -145,7 +145,7 @@
     "resnet": "hf-internal-testing/tiny-random-resnet",
     "roberta": "hf-internal-testing/tiny-random-RobertaModel",
     "roformer": "hf-internal-testing/tiny-random-RoFormerModel",
-    "rtdetr": "PekingU/rtdetr_r18vd",
+    "rt-detr": "PekingU/rtdetr_r18vd",
     "sam": "fxmarty/sam-vit-tiny-random",
     "segformer": "hf-internal-testing/tiny-random-SegformerModel",
     "splinter": "hf-internal-testing/tiny-random-SplinterModel",
@@ -265,7 +265,7 @@
     "resnet": "microsoft/resnet-50",
     "roberta": "roberta-base",
     "roformer": "junnyu/roformer_chinese_base",
-    "rtdetr": "PekingU/rtdetr_r101vd",
+    "rt-detr": "PekingU/rtdetr_r101vd",
     "sam": "facebook/sam-vit-base",
     "segformer": "nvidia/segformer-b0-finetuned-ade-512-512",
     "splinter": "hf-internal-testing/tiny-random-SplinterModel",