diff --git a/inference/models/aliases.py b/inference/models/aliases.py index aad879ffe1..390a017107 100644 --- a/inference/models/aliases.py +++ b/inference/models/aliases.py @@ -65,6 +65,13 @@ "rfdetr-seg-preview": "coco-dataset-vdnr1/26", } +CLASSIFICATION_ALIASES = { + "resent18": "classifiers/1", + "resent34": "classifiers/2", + "resent50": "classifiers/3", + "resent101": "classifiers/4", +} + REGISTERED_ALIASES = { "yolov8n-640": "coco/3", "yolov8n-1280": "coco/9", @@ -106,6 +113,7 @@ **YOLOV11_ALIASES, **QWEN_ALIASES, **RFDETR_ALIASES, + **CLASSIFICATION_ALIASES, } diff --git a/inference_sdk/http/utils/aliases.py b/inference_sdk/http/utils/aliases.py index 7af5a9b78e..977875d710 100644 --- a/inference_sdk/http/utils/aliases.py +++ b/inference_sdk/http/utils/aliases.py @@ -34,6 +34,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", @@ -92,6 +99,7 @@ **YOLOV11_ALIASES, **QWEN_ALIASES, **RFDETR_ALIASES, + **CLASSIFICATION_ALIASES, } OCR_ENDPOINTS = {