Skip to content

Commit f85c2a5

Browse files
Add 'rfdetr-seg-preview' to supported model types (#414)
* Add 'rfdetr-seg-preview' to supported model types * fix(pre_commit): 🎨 auto format pre-commit hooks --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
1 parent ec3710d commit f85c2a5

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

roboflow/util/model_processor.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -226,7 +226,14 @@ 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", "rfdetr-nano", "rfdetr-small", "rfdetr-medium"]
229+
_supported_types = [
230+
"rfdetr-base",
231+
"rfdetr-large",
232+
"rfdetr-nano",
233+
"rfdetr-small",
234+
"rfdetr-medium",
235+
"rfdetr-seg-preview",
236+
]
230237
if model_type not in _supported_types:
231238
raise ValueError(f"Model type {model_type} not supported. Supported types are {_supported_types}")
232239

0 commit comments

Comments
 (0)