File size: 809 Bytes
3d3d712 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
# Embedding
In TaskWeaver, we support various embedding models to generate embeddings for auto plugin selection.
## Embedding Configration
- `llm.embedding_api_type`: The type of the embedding API. We support the following types:
- openai
- qwen
- ollama
- sentence_transformers
- `llm.embedding_model`: The embedding model name. The model name should be aligned with `llm.embedding_api_type`.
We only list some embedding models we have tested below:
- openai
- text-embedding-ada-002
- qwen
- text-embedding-v1
- ollama
- llama2
- sentence_transformers
- all-mpnet-base-v2
- multi-qa-mpnet-base-dot-v1
- all-distilroberta-v1
- all-MiniLM-L12-v2
- multi-qa-MiniLM-L6-cos-v1
You also can use other embedding models supported by the above embedding APIs. |