From 1f71fae6eece83c8962c7c1e22906b9882ad46b3 Mon Sep 17 00:00:00 2001 From: Ella Charlaix Date: Tue, 17 Dec 2024 15:06:28 +0100 Subject: [PATCH] Update tests large model --- tests/exporters/exporters_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/exporters/exporters_utils.py b/tests/exporters/exporters_utils.py index 222821850b..bdbc73912c 100644 --- a/tests/exporters/exporters_utils.py +++ b/tests/exporters/exporters_utils.py @@ -265,7 +265,7 @@ "resnet": "microsoft/resnet-50", "roberta": "roberta-base", "roformer": "junnyu/roformer_chinese_base", - "rtdetr": "PekingU/rtdetr_r18vd", + "rtdetr": "PekingU/rtdetr_r101vd", "sam": "facebook/sam-vit-base", "segformer": "nvidia/segformer-b0-finetuned-ade-512-512", "splinter": "hf-internal-testing/tiny-random-SplinterModel",