Skip to content

Commit 9e92b03

Browse files
committed
Added missing Titan text embedding models
1 parent 6058492 commit 9e92b03

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

scrapegraphai/helpers/models_tokens.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -84,6 +84,9 @@
8484
"mistral.mistral-7b-instruct-v0:2": 32768,
8585
"mistral.mixtral-8x7b-instruct-v0:1": 32768,
8686
"mistral.mistral-large-2402-v1:0": 32768,
87+
# Embedding models
88+
"amazon.titan-embed-text-v1": 8000,
89+
"amazon.titan-embed-text-v2:0": 8000,
8790
"cohere.embed-english-v3": 512,
8891
"cohere.embed-multilingual-v3": 512
8992
},

0 commit comments

Comments
 (0)