Skip to content

Commit 10ea6b1

Browse files
committed
tests: Include RFDETRXLarge and RFDETR2XLarge models to instantiation tests
1 parent 7321185 commit 10ea6b1

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

tests/try_instantiate_all_models.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,10 +16,12 @@
1616
"""
1717

1818
import sys
19+
from functools import partial
1920

2021
from tqdm.auto import tqdm
2122

2223
from rfdetr import (
24+
RFDETR2XLarge,
2325
RFDETRBase,
2426
RFDETRLarge,
2527
RFDETRMedium,
@@ -32,6 +34,7 @@
3234
RFDETRSegSmall,
3335
RFDETRSegXLarge,
3436
RFDETRSmall,
37+
RFDETRXLarge,
3538
)
3639

3740
# Explicitly list all models to validate
@@ -42,6 +45,8 @@
4245
RFDETRMedium,
4346
RFDETRBase,
4447
RFDETRLarge,
48+
partial(RFDETRXLarge, accept_platform_model_license=True),
49+
partial(RFDETR2XLarge, accept_platform_model_license=True),
4550
# Segmentation Models
4651
RFDETRSegPreview,
4752
RFDETRSegNano,

0 commit comments

Comments
 (0)