diff --git a/roboflow/util/model_processor.py b/roboflow/util/model_processor.py index 5d469978..d7622d51 100644 --- a/roboflow/util/model_processor.py +++ b/roboflow/util/model_processor.py @@ -226,7 +226,14 @@ def _process_yolo(model_type: str, model_path: str, filename: str) -> str: def _process_rfdetr(model_type: str, model_path: str, filename: str) -> str: - _supported_types = ["rfdetr-base", "rfdetr-large", "rfdetr-nano", "rfdetr-small", "rfdetr-medium"] + _supported_types = [ + "rfdetr-base", + "rfdetr-large", + "rfdetr-nano", + "rfdetr-small", + "rfdetr-medium", + "rfdetr-seg-preview", + ] if model_type not in _supported_types: raise ValueError(f"Model type {model_type} not supported. Supported types are {_supported_types}")