Skip to content

Commit

Permalink
chore(model-hub): add embedding models (#1091)
Browse files Browse the repository at this point in the history
Because

- we have added support for embedding task

This commit

- add embedding models to model-hub
  • Loading branch information
heiruwu committed Sep 9, 2024
1 parent 8e9e76c commit bd15fdd
Showing 1 changed file with 45 additions and 6 deletions.
51 changes: 45 additions & 6 deletions model-hub/model_hub_gpu.json
Original file line number Diff line number Diff line change
Expand Up @@ -45,10 +45,10 @@
"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",
"task": "TASK_VISUAL_QUESTION_ANSWERING",
"task": "TASK_CHAT",
"region": "REGION_GCP_EUROPE_WEST4",
"hardware": "NVIDIA_L4",
"version": "v0.1.1",
"version": "v0.1.0",
"configuration": {}
},
{
Expand All @@ -71,7 +71,7 @@
"description": "Llama2-7b-Chat, from meta, is trained to generate text based on your prompts.",
"model_definition": "model-definitions/container",
"visibility": "VISIBILITY_PUBLIC",
"task": "TASK_TEXT_GENERATION_CHAT",
"task": "TASK_CHAT",
"region": "REGION_GCP_EUROPE_WEST4",
"hardware": "NVIDIA_L4",
"version": "v0.1.0",
Expand All @@ -84,7 +84,7 @@
"description": "Llama3-8b-Instruct, from meta, is trained to generate text based on your prompts.",
"model_definition": "model-definitions/container",
"visibility": "VISIBILITY_PUBLIC",
"task": "TASK_TEXT_GENERATION_CHAT",
"task": "TASK_CHAT",
"region": "REGION_GCP_EUROPE_WEST4",
"hardware": "NVIDIA_L4",
"version": "v0.1.0",
Expand All @@ -97,7 +97,7 @@
"description": "Zephyr-7b, from Huggingface, is trained to generate text based on your prompts.",
"model_definition": "model-definitions/container",
"visibility": "VISIBILITY_PUBLIC",
"task": "TASK_TEXT_GENERATION_CHAT",
"task": "TASK_CHAT",
"region": "REGION_GCP_EUROPE_WEST4",
"hardware": "NVIDIA_L4",
"version": "v0.1.0",
Expand All @@ -110,10 +110,49 @@
"description": "Llamacode-7b, from Huggingface, is trained to generate text based on your prompts.",
"model_definition": "model-definitions/container",
"visibility": "VISIBILITY_PUBLIC",
"task": "TASK_TEXT_GENERATION",
"task": "TASK_COMPLETION",
"region": "REGION_GCP_EUROPE_WEST4",
"hardware": "NVIDIA_L4",
"version": "v0.1.0",
"configuration": {}
},
{
"id": "gte-qwen2-1.5b-instruct",
"owner_type": "organizations",
"owner_id": "instill-ai",
"description": "gte-Qwen2-1.5B-instruct is the latest model in the gte (General Text Embedding) model family.",
"model_definition": "model-definitions/container",
"visibility": "VISIBILITY_PUBLIC",
"task": "TASK_EMBEDDING",
"region": "REGION_GCP_EUROPE_WEST4",
"hardware": "NVIDIA_T4",
"version": "v0.1.0",
"configuration": {}
},
{
"id": "jina-clip-v1",
"owner_type": "organizations",
"owner_id": "instill-ai",
"description": "jina-clip-v1 is a state-of-the-art English multimodal (text-image) embedding model.",
"model_definition": "model-definitions/container",
"visibility": "VISIBILITY_PUBLIC",
"task": "TASK_EMBEDDING",
"region": "REGION_GCP_EUROPE_WEST4",
"hardware": "NVIDIA_T4",
"version": "v0.1.0",
"configuration": {}
},
{
"id": "stella-en-1.5b-v5",
"owner_type": "organizations",
"owner_id": "instill-ai",
"description": "stella-en-1.5B-v5 is trained based on Alibaba-NLP/gte-large-en-v1.5 and Alibaba-NLP/gte-Qwen2-1.5B-instruct.",
"model_definition": "model-definitions/container",
"visibility": "VISIBILITY_PUBLIC",
"task": "TASK_EMBEDDING",
"region": "REGION_GCP_EUROPE_WEST4",
"hardware": "NVIDIA_T4",
"version": "v0.1.0",
"configuration": {}
}
]

0 comments on commit bd15fdd

Please sign in to comment.