diff --git a/tests/exporters/exporters_utils.py b/tests/exporters/exporters_utils.py index bdbc73912c..680ae7e91b 100644 --- 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",