[Feature] Add support for NVIDIA AI LLMs and embedding models (#1293)

2024-02-29 23:56:25 -08:00
parent 6518c0c06b
commit c77a75dfb5
18 changed files with 195 additions and 22 deletions
--- a/embedchain/models/vector_dimensions.py
+++ b/embedchain/models/vector_dimensions.py
@@ -9,3 +9,4 @@ class VectorDimensions(Enum):
    HUGGING_FACE = 384
    GOOGLE_AI = 768
    MISTRAL_AI = 1024
+    NVIDIA_AI = 1024