From dcd4d0c4fccd4b9b46a99603884cc37bf974050f Mon Sep 17 00:00:00 2001 From: Peter Robicheaux Date: Wed, 23 Jul 2025 04:08:07 -0700 Subject: [PATCH] Increase supported model types --- roboflow/util/model_processor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/roboflow/util/model_processor.py b/roboflow/util/model_processor.py index 78227b80..5d469978 100644 --- a/roboflow/util/model_processor.py +++ b/roboflow/util/model_processor.py @@ -226,7 +226,7 @@ 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"] + _supported_types = ["rfdetr-base", "rfdetr-large", "rfdetr-nano", "rfdetr-small", "rfdetr-medium"] if model_type not in _supported_types: raise ValueError(f"Model type {model_type} not supported. Supported types are {_supported_types}")