Skip to content
Merged
Show file tree
Hide file tree
Changes from 5 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 8 additions & 0 deletions inference/models/aliases.py
Original file line number Diff line number Diff line change
Expand Up @@ -64,6 +64,13 @@
"rfdetr-medium": "coco/40",
}

CLASSIFICATION_ALIASES = {
"resent18": "classifiers/1",
"resent34": "classifiers/2",
"resent50": "classifiers/3",
"resent101": "classifiers/4",
}

REGISTERED_ALIASES = {
"yolov8n-640": "coco/3",
"yolov8n-1280": "coco/9",
Expand Down Expand Up @@ -105,6 +112,7 @@
**YOLOV11_ALIASES,
**QWEN_ALIASES,
**RFDETR_ALIASES,
**CLASSIFICATION_ALIASES,
}


Expand Down
8 changes: 8 additions & 0 deletions inference_sdk/http/utils/aliases.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,13 @@
"qwen25-vl-7b": "qwen-pretrains/1",
}

CLASSIFICATION_ALIASES = {
"resent18": "classifiers/1",
"resent34": "classifiers/2",
"resent50": "classifiers/3",
"resent101": "classifiers/4",
}

REGISTERED_ALIASES = {
"yolov8n-640": "coco/3",
"yolov8n-1280": "coco/9",
Expand Down Expand Up @@ -91,6 +98,7 @@
**YOLOV11_ALIASES,
**QWEN_ALIASES,
**RFDETR_ALIASES,
**CLASSIFICATION_ALIASES,
}

OCR_ENDPOINTS = {
Expand Down