vector_dimensions.py 284 B

1234567891011121314
  1. from enum import Enum
  2. # vector length created by embedding fn
  3. class VectorDimensions(Enum):
  4. GPT4ALL = 384
  5. OPENAI = 1536
  6. VERTEX_AI = 768
  7. HUGGING_FACE = 384
  8. GOOGLE_AI = 768
  9. MISTRAL_AI = 1024
  10. NVIDIA_AI = 1024
  11. COHERE = 384
  12. OLLAMA = 384