File tree Expand file tree Collapse file tree
inference_models/inference_models/models/auto_loaders Expand file tree Collapse file tree Original file line number Diff line number Diff 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" ,
You can’t perform that action at this time.
0 commit comments