Skip to content

Commit

Permalink
feat(model): support Llava 13b (#442)
Browse files Browse the repository at this point in the history
Because

- support llava 13b

This commit

- replace llava 7b with llava 13b

todo
- update [config
here](https://github.com/instill-ai/terraform-instill-cloud-gcp-services/blob/c1b82fb422a62baad1fe0b2e8c190c45c8b8c07b/terraform-instill-core/main.tf#L166)
once the pr merged
  • Loading branch information
tonywang10101 committed Mar 12, 2024
1 parent c1a7288 commit 39bbf8c
Showing 1 changed file with 92 additions and 0 deletions.
92 changes: 92 additions & 0 deletions model-hub/model_hub_gpu.json
@@ -0,0 +1,92 @@
[
{
"id": "mobilenetv2",
"description": "An efficient image classification model, pretrained on ImageNet dataset which contains images from 1,000 classes.",
"task": "TASK_CLASSIFICATION",
"model_definition": "model-definitions/github",
"configuration": {
"repository": "instill-ai/model-mobilenetv2-dvc",
"tag": "v0.7.0-ray-gpu"
}
},
{
"id": "yolov7",
"description": "YOLOv7 is a state-of-the-art real-time object detector pretrained on MS COCO dataset with 80 object classes.",
"task": "TASK_DETECTION",
"model_definition": "model-definitions/github",
"configuration": {
"repository": "instill-ai/model-yolov7-dvc",
"tag": "v0.7.0-ray-gpu"
}
},
{
"id": "yolov7-stomata",
"description": "The instance segmentation model based on Yolov7, fine-tuned on the custom dataset with 337 images and 1 object class 'stomata'.",
"task": "TASK_INSTANCE_SEGMENTATION",
"model_definition": "model-definitions/github",
"configuration": {
"repository": "instill-ai/model-stomata-instance-segmentation-dvc",
"tag": "v0.7.0-yolov7-mask-ray-gpu"
}
},
{
"id": "llama2-7b",
"description": "Llama2-7b, from meta, is trained to generate text based on your prompts.",
"task": "TASK_TEXT_GENERATION",
"model_definition": "model-definitions/github",
"configuration": {
"repository": "instill-ai/model-llama2-7b-dvc",
"tag": "f32-cpu-transformer-ray-v0.8.0"
}
},
{
"id": "llama2-7b-chat",
"description": "Llama2-7b-Chat, from meta, is trained to generate text based on your prompts.",
"task": "TASK_TEXT_GENERATION_CHAT",
"model_definition": "model-definitions/github",
"configuration": {
"repository": "instill-ai/model-llama2-7b-chat-dvc",
"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.",
"task": "TASK_TEXT_GENERATION_CHAT",
"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"
}
},
{
"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"
}
}
]

0 comments on commit 39bbf8c

Please sign in to comment.