From c9d27b6852ce6e8b8257f309f0624be9ffc44fcd Mon Sep 17 00:00:00 2001 From: HR Wu <5631010+heiruwu@users.noreply.github.com> Date: Wed, 5 Jun 2024 03:08:46 +0800 Subject: [PATCH] chore(model-hub): update model-hub configs (#1009) Because - Predeploy model lists has models which will be deployed under different namespace This commit - add namespace id and type in model-hub --- charts/core/values.yaml | 2 - model-hub/model_hub_gpu.json | 18 ++++++ model-hub/model_hub_test.json | 100 ++++++++++++++++++++-------------- 3 files changed, 78 insertions(+), 42 deletions(-) diff --git a/charts/core/values.yaml b/charts/core/values.yaml index a58482db..fe9c26b1 100644 --- a/charts/core/values.yaml +++ b/charts/core/values.yaml @@ -496,8 +496,6 @@ modelBackend: replicaCount: 1 # -- The model initialization configuration initModel: - ownertype: users - ownerid: admin enabled: false path: https://raw.githubusercontent.com/instill-ai/instill-core/main/model-hub/model_hub_test.json # -- Add extra env variables diff --git a/model-hub/model_hub_gpu.json b/model-hub/model_hub_gpu.json index 22a96941..1de02425 100644 --- a/model-hub/model_hub_gpu.json +++ b/model-hub/model_hub_gpu.json @@ -1,6 +1,8 @@ [ { "id": "mobilenetv2", + "owner_type": "organizations", + "owner_id": "instill-ai", "description": "An efficient image classification model, pretrained on ImageNet dataset which contains images from 1,000 classes.", "model_definition": "model-definitions/container", "visibility": "VISIBILITY_PUBLIC", @@ -12,6 +14,8 @@ }, { "id": "yolov7", + "owner_type": "organizations", + "owner_id": "instill-ai", "description": "YOLOv7 is a state-of-the-art real-time object detector pretrained on MS COCO dataset with 80 object classes.", "model_definition": "model-definitions/container", "visibility": "VISIBILITY_PUBLIC", @@ -23,6 +27,8 @@ }, { "id": "yolov7-stomata", + "owner_type": "organizations", + "owner_id": "abrc", "description": "The instance segmentation model based on Yolov7, fine-tuned on the custom dataset with 337 images and 1 object class 'stomata'.", "model_definition": "model-definitions/container", "visibility": "VISIBILITY_PUBLIC", @@ -34,6 +40,8 @@ }, { "id": "llava-1-6-13b", + "owner_type": "organizations", + "owner_id": "instill-ai", "description": "LLaVa-13b, from liuhaotian, is trained to generate text based on your prompts with miltimodal input.", "model_definition": "model-definitions/container", "visibility": "VISIBILITY_PUBLIC", @@ -45,6 +53,8 @@ }, { "id": "stable-diffusion-xl", + "owner_type": "organizations", + "owner_id": "instill-ai", "description": "Stable-Diffusion-XL, from StabilityAI, is trained to generate image based on your prompts.", "model_definition": "model-definitions/container", "visibility": "VISIBILITY_PUBLIC", @@ -56,6 +66,8 @@ }, { "id": "llama2-7b-chat", + "owner_type": "organizations", + "owner_id": "instill-ai", "description": "Llama2-7b-Chat, from meta, is trained to generate text based on your prompts.", "model_definition": "model-definitions/container", "visibility": "VISIBILITY_PUBLIC", @@ -67,6 +79,8 @@ }, { "id": "llama3-8b-instruct", + "owner_type": "organizations", + "owner_id": "instill-ai", "description": "Llama3-8b-Instruct, from meta, is trained to generate text based on your prompts.", "model_definition": "model-definitions/container", "visibility": "VISIBILITY_PUBLIC", @@ -78,6 +92,8 @@ }, { "id": "zephyr-7b", + "owner_type": "organizations", + "owner_id": "instill-ai", "description": "Zephyr-7b, from Huggingface, is trained to generate text based on your prompts.", "model_definition": "model-definitions/container", "visibility": "VISIBILITY_PUBLIC", @@ -89,6 +105,8 @@ }, { "id": "llamacode-7b", + "owner_type": "organizations", + "owner_id": "instill-ai", "description": "Llamacode-7b, from Huggingface, is trained to generate text based on your prompts.", "model_definition": "model-definitions/container", "visibility": "VISIBILITY_PUBLIC", diff --git a/model-hub/model_hub_test.json b/model-hub/model_hub_test.json index 765f73a5..72e26189 100644 --- a/model-hub/model_hub_test.json +++ b/model-hub/model_hub_test.json @@ -1,112 +1,132 @@ [ { "id": "dummy-cls", + "owner_type": "users", + "owner_id": "admin", "description": "A classification model for integration-test", "model_definition": "model-definitions/container", "visibility": "VISIBILITY_PUBLIC", - "region": "REGION_GCP_EUROPE_WEST_4", + "task": "TASK_CLASSIFICATION", + "region": "REGION_LOCAL", "hardware": "CPU", - "configuration": { - "task": "CLASSIFICATION" - } + "version": "latest", + "configuration": {} }, { "id": "dummy-det", + "owner_type": "users", + "owner_id": "admin", "description": "A object detection model for integration-test", "model_definition": "model-definitions/container", "visibility": "VISIBILITY_PUBLIC", - "region": "REGION_GCP_EUROPE_WEST_4", + "task": "TASK_DETECTION", + "region": "REGION_LOCAL", "hardware": "CPU", - "configuration": { - "task": "DETECTION" - } + "version": "latest", + "configuration": {} }, { "id": "dummy-image-to-image", + "owner_type": "users", + "owner_id": "admin", "description": "A image to image model for integration-test", "model_definition": "model-definitions/container", "visibility": "VISIBILITY_PUBLIC", - "region": "REGION_GCP_EUROPE_WEST_4", + "task": "TASK_IMAGE_TO_IMAGE", + "region": "REGION_LOCAL", "hardware": "CPU", - "configuration": { - "task": "IMAGE_TO_IMAGE" - } + "version": "latest", + "configuration": {} }, { "id": "dummy-instance-segmentation", + "owner_type": "users", + "owner_id": "admin", "description": "A instance segmentation model for integration-test", "model_definition": "model-definitions/container", "visibility": "VISIBILITY_PUBLIC", - "region": "REGION_GCP_EUROPE_WEST_4", + "task": "TASK_INSTANCE_SEGMENTATION", + "region": "REGION_LOCAL", "hardware": "CPU", - "configuration": { - "task": "INSTANCE_SEGMENTATION" - } + "version": "latest", + "configuration": {} }, { "id": "dummy-keypoint", + "owner_type": "users", + "owner_id": "admin", "description": "A keypoint detection model for integration-test", "model_definition": "model-definitions/container", "visibility": "VISIBILITY_PUBLIC", - "region": "REGION_GCP_EUROPE_WEST_4", + "task": "TASK_KEYPOINT", + "region": "REGION_LOCAL", "hardware": "CPU", - "configuration": { - "task": "KEYPOINT" - } + "version": "latest", + "configuration": {} }, { "id": "dummy-semantic-segmentation", + "owner_type": "users", + "owner_id": "admin", "description": "A semantic segmentation model for integration-test", "model_definition": "model-definitions/container", "visibility": "VISIBILITY_PUBLIC", - "region": "REGION_GCP_EUROPE_WEST_4", + "region": "REGION_LOCAL", "hardware": "CPU", - "configuration": { - "task": "SEMANTIC_SEGMENTATION" - } + "task": "TASK_SEMANTIC_SEGMENTATION", + "version": "latest", + "configuration": {} }, { "id": "dummy-text-generation", + "owner_type": "users", + "owner_id": "admin", "description": "A text generation model for integration-test", "model_definition": "model-definitions/container", "visibility": "VISIBILITY_PUBLIC", - "region": "REGION_GCP_EUROPE_WEST_4", + "task": "TASK_TEXT_GENERATION", + "region": "REGION_LOCAL", "hardware": "CPU", - "configuration": { - "task": "TEXT_GENERATION" - } + "version": "latest", + "configuration": {} }, { "id": "dummy-text-generation-chat", + "owner_type": "users", + "owner_id": "admin", "description": "A text generation chat model for integration-test", "model_definition": "model-definitions/container", "visibility": "VISIBILITY_PUBLIC", - "region": "REGION_GCP_EUROPE_WEST_4", + "task": "TASK_TEXT_GENERATION_CHAT", + "region": "REGION_LOCAL", "hardware": "CPU", - "configuration": { - "task": "TEXT_GENERATION_CHAT" - } + "version": "latest", + "configuration": {} }, { "id": "dummy-text-to-image", + "owner_type": "users", + "owner_id": "admin", "description": "A text to image model for integration-test", "model_definition": "model-definitions/container", "visibility": "VISIBILITY_PUBLIC", - "region": "REGION_GCP_EUROPE_WEST_4", + "task": "TASK_TEXT_TO_IMAGE", + "region": "REGION_LOCAL", "hardware": "CPU", - "configuration": { - "task": "TEXT_TO_IMAGE" - } + "version": "latest", + "configuration": {} }, { "id": "dummy-visual-question-answering", + "owner_type": "users", + "owner_id": "admin", "description": "A visual question answering model for integration-test", "model_definition": "model-definitions/container", "visibility": "VISIBILITY_PUBLIC", - "region": "REGION_GCP_EUROPE_WEST_4", + "task": "TASK_VISUAL_QUESTION_ANSWERING", + "region": "REGION_LOCAL", "hardware": "CPU", - "configuration": { - "task": "VISUAL_QUESTION_ANSWERING" - } + "version": "latest", + "configuration": {} } ]