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
instill-ai · Jun 4, 2024 · c9d27b6 · c9d27b6
1 parent 2130401
commit c9d27b6
Show file tree

Hide file tree

Showing 3 changed files with 78 additions and 42 deletions.
diff --git 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
@@ -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
@@ -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": {}
   }
 ]