server : be explicit about the pooling type in the tests

ggml-ci
This commit is contained in:
Georgi Gerganov 2024-12-17 11:45:18 +02:00
parent 06e85401b0
commit 1b18b2d7b0
No known key found for this signature in database
GPG key ID: 449E073F9DC10735

View file

@ -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