server : be explicit about the pooling type in the tests
ggml-ci
This commit is contained in:
parent
06e85401b0
commit
1b18b2d7b0
1 changed files with 1 additions and 2 deletions
|
@ -275,7 +275,7 @@ class ServerPreset:
|
|||
return server
|
||||
|
||||
@staticmethod
|
||||
def bert_bge_small(pooling = 'last') -> ServerProcess:
|
||||
def bert_bge_small() -> ServerProcess:
|
||||
server = ServerProcess()
|
||||
server.model_hf_repo = "ggml-org/models"
|
||||
server.model_hf_file = "bert-bge-small/ggml-model-f16.gguf"
|
||||
|
@ -286,7 +286,6 @@ class ServerPreset:
|
|||
server.n_slots = 2
|
||||
server.seed = 42
|
||||
server.server_embeddings = True
|
||||
server.pooling = pooling
|
||||
return server
|
||||
|
||||
@staticmethod
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue