Skip to content

Commit dcd4d0c

Browse files
committed
Increase supported model types
1 parent 363b116 commit dcd4d0c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

roboflow/util/model_processor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -226,7 +226,7 @@ def _process_yolo(model_type: str, model_path: str, filename: str) -> str:
226226

227227

228228
def _process_rfdetr(model_type: str, model_path: str, filename: str) -> str:
229-
_supported_types = ["rfdetr-base", "rfdetr-large"]
229+
_supported_types = ["rfdetr-base", "rfdetr-large", "rfdetr-nano", "rfdetr-small", "rfdetr-medium"]
230230
if model_type not in _supported_types:
231231
raise ValueError(f"Model type {model_type} not supported. Supported types are {_supported_types}")
232232

0 commit comments

Comments
 (0)