From 60948583d281049287082c076fb4106c04ea8240 Mon Sep 17 00:00:00 2001
From: "tony.wang.10101" <tony.wang.10101@gmail.com>
Date: Sun, 24 Mar 2024 19:42:10 +0800
Subject: [PATCH] feat(model): support new models for ver 0.12.0

---
 model-hub/model_hub_gpu.json | 62 +++++++++++++++++++++---------------
 1 file changed, 36 insertions(+), 26 deletions(-)

diff --git a/model-hub/model_hub_gpu.json b/model-hub/model_hub_gpu.json
index 25cbeccb..4fd45416 100644
--- a/model-hub/model_hub_gpu.json
+++ b/model-hub/model_hub_gpu.json
@@ -29,6 +29,36 @@
       "tag": "v0.7.0-yolov7-mask-ray-gpu"
     }
   },
+  {
+    "id": "llava-1-6-13b",
+    "description": "LLaVa-13b, from liuhaotian, is trained to generate text based on your prompts with miltimodal input.",
+    "task": "TASK_VISUAL_QUESTION_ANSWERING",
+    "model_definition": "model-definitions/github",
+    "configuration": {
+      "repository": "instill-ai/model-llava-13b-dvc",
+      "tag": "f16-gpuAuto-transformer-ray-v0.11.0"
+    }
+  },
+  {
+    "id": "stable-diffusion-xl",
+    "description": "Stable-Diffusion-XL, from StabilityAI, is trained to generate image based on your prompts.",
+    "task": "TASK_TEXT_TO_IMAGE",
+    "model_definition": "model-definitions/github",
+    "configuration": {
+      "repository": "instill-ai/model-diffusion-xl-dvc",
+      "tag": "f16-gpuAuto-diffusers-ray-v0.8.0"
+    }
+  },
+  {
+    "id": "controlnet-canny",
+    "description": "ControlNet-Canny Version, from Lvmin, is trained to generate image based on your prompts and images.",
+    "task": "TASK_IMAGE_TO_IMAGE",
+    "model_definition": "model-definitions/github",
+    "configuration": {
+      "repository": "instill-ai/model-controlnet-dvc",
+      "tag": "f16-gpuAuto-diffusers-ray-v0.8.0"
+    }
+  },
   {
     "id": "llama2-7b",
     "description": "Llama2-7b, from meta, is trained to generate text based on your prompts.",
@@ -49,16 +79,6 @@
       "tag": "f16-gpuAuto-transformer-ray-v0.8.0"
     }
   },
-  {
-    "id": "llava-1-6-13b",
-    "description": "LLaVa-13b, from liuhaotian, is trained to generate text based on your prompts with miltimodal input.",
-    "task": "TASK_VISUAL_QUESTION_ANSWERING",
-    "model_definition": "model-definitions/github",
-    "configuration": {
-      "repository": "instill-ai/model-llava-13b-dvc",
-      "tag": "f16-gpuAuto-transformer-ray-v0.11.0"
-    }
-  },
   {
     "id": "zephyr-7b",
     "description": "Zephyr-7b, from Huggingface, is trained to generate text based on your prompts.",
@@ -66,27 +86,17 @@
     "model_definition": "model-definitions/github",
     "configuration": {
       "repository": "instill-ai/model-zephyr-7b-dvc",
-      "tag": "f32-cpu-transformer-ray-v0.8.0"
-    }
-  },
-  {
-    "id": "stable-diffusion-xl",
-    "description": "Stable-Diffusion-XL, from StabilityAI, is trained to generate image based on your prompts.",
-    "task": "TASK_TEXT_TO_IMAGE",
-    "model_definition": "model-definitions/github",
-    "configuration": {
-      "repository": "instill-ai/model-diffusion-xl-dvc",
-      "tag": "f16-gpuAuto-diffusers-ray-v0.8.0"
+      "tag": "f16-1gpu-transformer-ray-v0.11.0"
     }
   },
   {
-    "id": "controlnet-canny",
-    "description": "ControlNet-Canny Version, from Lvmin, is trained to generate image based on your prompts and images.",
-    "task": "TASK_IMAGE_TO_IMAGE",
+    "id": "llamacode-7b",
+    "description": "Llamacode-7b, from Huggingface, is trained to generate text based on your prompts.",
+    "task": "TASK_TEXT_GENERATION_CHAT",
     "model_definition": "model-definitions/github",
     "configuration": {
-      "repository": "instill-ai/model-controlnet-dvc",
-      "tag": "f16-gpuAuto-diffusers-ray-v0.8.0"
+      "repository": "instill-ai/model-codellama-7b-dvc",
+      "tag": "f16-1gpu-transformer-ray-v0.11.0"
     }
   }
 ]