chore(model_hub): add test model list for integration-test (#483)

Because - We want to define a list of test models for integration-test This commit - add a model list for integration-test
instill-ai · Apr 17, 2024 · 1afe638 · 1afe638
1 parent b66c410
commit 1afe638
Showing 1 changed file with 112 additions and 0 deletions.
diff --git a/model-hub/model_hub_test.json b/model-hub/model_hub_test.json
@@ -0,0 +1,112 @@
+[
+  {
+    "id": "dummy-cls",
+    "description": "A classification model for integration-test",
+    "model_definition": "model-definitions/container",
+    "visibility": "VISIBILITY_PUBLIC",
+    "region": "REGION_GCP_EUROPE_WEST_4",
+    "hardware": "CPU",
+    "configuration": {
+      "task": "CLASSIFICATION"
+    }
+  },
+  {
+    "id": "dummy-det",
+    "description": "A object detection model for integration-test",
+    "model_definition": "model-definitions/container",
+    "visibility": "VISIBILITY_PUBLIC",
+    "region": "REGION_GCP_EUROPE_WEST_4",
+    "hardware": "CPU",
+    "configuration": {
+      "task": "DETECTION"
+    }
+  },
+  {
+    "id": "dummy-image-to-image",
+    "description": "A image to image model for integration-test",
+    "model_definition": "model-definitions/container",
+    "visibility": "VISIBILITY_PUBLIC",
+    "region": "REGION_GCP_EUROPE_WEST_4",
+    "hardware": "CPU",
+    "configuration": {
+      "task": "IMAGE_TO_IMAGE"
+    }
+  },
+  {
+    "id": "dummy-instance-segmentation",
+    "description": "A instance segmentation model for integration-test",
+    "model_definition": "model-definitions/container",
+    "visibility": "VISIBILITY_PUBLIC",
+    "region": "REGION_GCP_EUROPE_WEST_4",
+    "hardware": "CPU",
+    "configuration": {
+      "task": "INSTANCE_SEGMENTATION"
+    }
+  },
+  {
+    "id": "dummy-keypoint",
+    "description": "A keypoint detection model for integration-test",
+    "model_definition": "model-definitions/container",
+    "visibility": "VISIBILITY_PUBLIC",
+    "region": "REGION_GCP_EUROPE_WEST_4",
+    "hardware": "CPU",
+    "configuration": {
+      "task": "KEYPOINT"
+    }
+  },
+  {
+    "id": "dummy-semantic-segmentation",
+    "description": "A semantic segmentation model for integration-test",
+    "model_definition": "model-definitions/container",
+    "visibility": "VISIBILITY_PUBLIC",
+    "region": "REGION_GCP_EUROPE_WEST_4",
+    "hardware": "CPU",
+    "configuration": {
+      "task": "SEMANTIC_SEGMENTATION"
+    }
+  },
+  {
+    "id": "dummy-text-generation",
+    "description": "A text generation model for integration-test",
+    "model_definition": "model-definitions/container",
+    "visibility": "VISIBILITY_PUBLIC",
+    "region": "REGION_GCP_EUROPE_WEST_4",
+    "hardware": "CPU",
+    "configuration": {
+      "task": "TEXT_GENERATION"
+    }
+  },
+  {
+    "id": "dummy-text-generation-chat",
+    "description": "A text generation chat model for integration-test",
+    "model_definition": "model-definitions/container",
+    "visibility": "VISIBILITY_PUBLIC",
+    "region": "REGION_GCP_EUROPE_WEST_4",
+    "hardware": "CPU",
+    "configuration": {
+      "task": "TEXT_GENERATION_CHAT"
+    }
+  },
+  {
+    "id": "dummy-text-to-image",
+    "description": "A text to image model for integration-test",
+    "model_definition": "model-definitions/container",
+    "visibility": "VISIBILITY_PUBLIC",
+    "region": "REGION_GCP_EUROPE_WEST_4",
+    "hardware": "CPU",
+    "configuration": {
+      "task": "TEXT_TO_IMAGE"
+    }
+  },
+  {
+    "id": "dummy-visual-question-answering",
+    "description": "A visual question answering model for integration-test",
+    "model_definition": "model-definitions/container",
+    "visibility": "VISIBILITY_PUBLIC",
+    "region": "REGION_GCP_EUROPE_WEST_4",
+    "hardware": "CPU",
+    "configuration": {
+      "task": "VISUAL_QUESTION_ANSWERING"
+    }
+  }
+]