AVB/bedrock-access-gateway
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add titan G1 embeddings (#152)

Browse Source
This commit is contained in:
Zack Elias
2025-06-17 11:09:22 +08:00
committed by GitHub
parent aed57307bc
commit 844efec086
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/api/models/bedrock.py
+2 -1
View File
@@ -68,8 +68,9 @@ cr_inference_prefix = get_inference_region_prefix()
SUPPORTED_BEDROCK_EMBEDDING_MODELS = { SUPPORTED_BEDROCK_EMBEDDING_MODELS = {
"cohere.embed-multilingual-v3": "Cohere Embed Multilingual", "cohere.embed-multilingual-v3": "Cohere Embed Multilingual",
"cohere.embed-english-v3": "Cohere Embed English", "cohere.embed-english-v3": "Cohere Embed English",
# Disable Titan embedding. "amazon.titan-embed-text-v1": "Titan Embeddings G1 - Text",
"amazon.titan-embed-text-v2:0": "Titan Embeddings G2 - Text", "amazon.titan-embed-text-v2:0": "Titan Embeddings G2 - Text",
# Disable Titan embedding.
# "amazon.titan-embed-image-v1": "Titan Multimodal Embeddings G1" # "amazon.titan-embed-image-v1": "Titan Multimodal Embeddings G1"
} }