mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-10-12 05:54:38 +00:00
refactor: switch to the new default nomic-embed-text-v1.5 embedding model in LS
This commit is contained in:
parent
f1748e2f92
commit
1d0f0a0d8e
63 changed files with 16170 additions and 186 deletions
|
@ -35,7 +35,7 @@ Model parameters can be influenced by the following options:
|
|||
- `--embedding-model`: comma-separated list of embedding models.
|
||||
- `--safety-shield`: comma-separated list of safety shields.
|
||||
- `--judge-model`: comma-separated list of judge models.
|
||||
- `--embedding-dimension`: output dimensionality of the embedding model to use for testing. Default: 384
|
||||
- `--embedding-dimension`: output dimensionality of the embedding model to use for testing. Default: 768
|
||||
|
||||
Each of these are comma-separated lists and can be used to generate multiple parameter combinations. Note that tests will be skipped
|
||||
if no model is specified.
|
||||
|
@ -82,7 +82,7 @@ OLLAMA_URL=http://localhost:11434 \
|
|||
pytest -s -v tests/integration/inference/test_text_inference.py \
|
||||
--stack-config=server:starter \
|
||||
--text-model=ollama/llama3.2:3b-instruct-fp16 \
|
||||
--embedding-model=sentence-transformers/all-MiniLM-L6-v2
|
||||
--embedding-model=nomic-embed-text-v1.5
|
||||
```
|
||||
|
||||
Run tests with auto-server startup on a custom port:
|
||||
|
@ -92,7 +92,7 @@ OLLAMA_URL=http://localhost:11434 \
|
|||
pytest -s -v tests/integration/inference/ \
|
||||
--stack-config=server:starter:8322 \
|
||||
--text-model=ollama/llama3.2:3b-instruct-fp16 \
|
||||
--embedding-model=sentence-transformers/all-MiniLM-L6-v2
|
||||
--embedding-model=nomic-embed-text-v1.5
|
||||
```
|
||||
|
||||
### Testing with Library Client
|
||||
|
@ -120,7 +120,7 @@ Another example: Running Vector IO tests for embedding models:
|
|||
```bash
|
||||
pytest -s -v tests/integration/vector_io/ \
|
||||
--stack-config=inference=inline::sentence-transformers,vector_io=inline::sqlite-vec \
|
||||
--embedding-model=sentence-transformers/all-MiniLM-L6-v2
|
||||
--embedding-model=nomic-embed-text-v1.5
|
||||
```
|
||||
|
||||
## Recording Modes
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue