Multi-task instructor model with quantization support & weak_lru cache for embedding function models (#612)

resolves #608
This commit is contained in:
Ayush Chaurasia
2023-11-09 12:34:18 +05:30
committed by Weston Pace
parent 2f2964a645
commit c0a49a9a5b
9 changed files with 270 additions and 10 deletions

View File

@@ -50,7 +50,7 @@ tests = ["pandas>=1.4", "pytest", "pytest-mock", "pytest-asyncio", "requests"]
dev = ["ruff", "pre-commit", "black"]
docs = ["mkdocs", "mkdocs-jupyter", "mkdocs-material", "mkdocstrings[python]"]
clip = ["torch", "pillow", "open-clip"]
embeddings = ["openai", "sentence-transformers", "torch", "pillow", "open-clip-torch", "cohere"]
embeddings = ["openai", "sentence-transformers", "torch", "pillow", "open-clip-torch", "cohere", "InstructorEmbedding"]
[build-system]
requires = ["setuptools", "wheel"]