mirror of
https://github.com/ggerganov/llama.cpp.git
synced 2024-11-11 21:39:52 +00:00
8843a98c2b
* args: default --model to models/ + filename from --model-url or --hf-file (or else legacy models/7B/ggml-model-f16.gguf) * args: main & server now call gpt_params_handle_model_default * args: define DEFAULT_MODEL_PATH + update cli docs * curl: check url of previous download (.json metadata w/ url, etag & lastModified) * args: fix update to quantize-stats.cpp * curl: support legacy .etag / .lastModified companion files * curl: rm legacy .etag file support * curl: reuse regex across headers callback calls * curl: unique_ptr to manage lifecycle of curl & outfile * curl: nit: no need for multiline regex flag * curl: update failed test (model file collision) + gitignore *.gguf.json
97 lines
2.4 KiB
Gherkin
97 lines
2.4 KiB
Gherkin
@llama.cpp
|
|
@embeddings
|
|
Feature: llama.cpp server
|
|
|
|
Background: Server startup
|
|
Given a server listening on localhost:8080
|
|
And a model url https://huggingface.co/ggml-org/models/resolve/main/bert-bge-small/ggml-model-f16.gguf
|
|
And a model file bert-bge-small.gguf
|
|
And a model alias bert-bge-small
|
|
And 42 as server seed
|
|
And 2 slots
|
|
And 1024 as batch size
|
|
And 1024 as ubatch size
|
|
And 2048 KV cache size
|
|
And embeddings extraction
|
|
Then the server is starting
|
|
Then the server is healthy
|
|
|
|
Scenario: Embedding
|
|
When embeddings are computed for:
|
|
"""
|
|
What is the capital of Bulgaria ?
|
|
"""
|
|
Then embeddings are generated
|
|
|
|
Scenario: OAI Embeddings compatibility
|
|
Given a model bert-bge-small
|
|
When an OAI compatible embeddings computation request for:
|
|
"""
|
|
What is the capital of Spain ?
|
|
"""
|
|
Then embeddings are generated
|
|
|
|
Scenario: OAI Embeddings compatibility with multiple inputs
|
|
Given a model bert-bge-small
|
|
Given a prompt:
|
|
"""
|
|
In which country Paris is located ?
|
|
"""
|
|
And a prompt:
|
|
"""
|
|
Is Madrid the capital of Spain ?
|
|
"""
|
|
When an OAI compatible embeddings computation request for multiple inputs
|
|
Then embeddings are generated
|
|
|
|
Scenario: Multi users embeddings
|
|
Given a prompt:
|
|
"""
|
|
Write a very long story about AI.
|
|
"""
|
|
And a prompt:
|
|
"""
|
|
Write another very long music lyrics.
|
|
"""
|
|
And a prompt:
|
|
"""
|
|
Write a very long poem.
|
|
"""
|
|
And a prompt:
|
|
"""
|
|
Write a very long joke.
|
|
"""
|
|
Given concurrent embedding requests
|
|
Then the server is busy
|
|
Then the server is idle
|
|
Then all embeddings are generated
|
|
|
|
Scenario: Multi users OAI compatibility embeddings
|
|
Given a prompt:
|
|
"""
|
|
In which country Paris is located ?
|
|
"""
|
|
And a prompt:
|
|
"""
|
|
Is Madrid the capital of Spain ?
|
|
"""
|
|
And a prompt:
|
|
"""
|
|
What is the biggest US city ?
|
|
"""
|
|
And a prompt:
|
|
"""
|
|
What is the capital of Bulgaria ?
|
|
"""
|
|
And a model bert-bge-small
|
|
Given concurrent OAI embedding requests
|
|
Then the server is busy
|
|
Then the server is idle
|
|
Then all embeddings are generated
|
|
|
|
Scenario: All embeddings should be the same
|
|
Given 10 fixed prompts
|
|
And a model bert-bge-small
|
|
Given concurrent OAI embedding requests
|
|
Then all embeddings are the same
|