Skip to content

Commit 327f774

Browse files
committed
Support rfdetr-nas-seg models
1 parent 1877762 commit 327f774

1 file changed

Lines changed: 13 additions & 3 deletions

File tree

inference_models/inference_models/models/auto_loaders/models_registry.py

Lines changed: 13 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -328,9 +328,19 @@ class RegistryEntry:
328328
module_name="inference_models.models.rfdetr.rfdetr_instance_segmentation_pytorch",
329329
class_name="RFDetrForInstanceSegmentationTorch",
330330
),
331-
("rfdetr", INSTANCE_SEGMENTATION_TASK, BackendType.ONNX): LazyClass(
332-
module_name="inference_models.models.rfdetr.rfdetr_instance_segmentation_onnx",
333-
class_name="RFDetrForInstanceSegmentationOnnx",
331+
("rfdetr", INSTANCE_SEGMENTATION_TASK, BackendType.ONNX): RegistryEntry(
332+
model_class=LazyClass(
333+
module_name="inference_models.models.rfdetr.rfdetr_instance_segmentation_onnx",
334+
class_name="RFDetrForInstanceSegmentationOnnx",
335+
),
336+
supported_model_features={
337+
"resolution",
338+
"patch_size",
339+
"num_windows",
340+
"dec_layers",
341+
"num_queries",
342+
"num_select",
343+
},
334344
),
335345
("rfdetr", INSTANCE_SEGMENTATION_TASK, BackendType.TRT): LazyClass(
336346
module_name="inference_models.models.rfdetr.rfdetr_instance_segmentation_trt",

0 commit comments

Comments
 (0)