Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat(model): support Llava 13b (#442)
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
1 parent
c1a7288
commit 39bbf8c
Showing
1 changed file
with
92 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -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" | ||
} | ||
} | ||
] |