From beccf0a336d0585a29e4017898b9758b13f2f37e Mon Sep 17 00:00:00 2001 From: Ettore Di Giacinto Date: Fri, 17 Jan 2025 19:27:15 +0100 Subject: [PATCH] Add alias to silently redirect sentencetransformers to transformers Signed-off-by: Ettore Di Giacinto --- pkg/model/initializers.go | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/pkg/model/initializers.go b/pkg/model/initializers.go index f4675050..0c964f2a 100644 --- a/pkg/model/initializers.go +++ b/pkg/model/initializers.go @@ -27,6 +27,11 @@ var Aliases map[string]string = map[string]string{ "embedded-store": LocalStoreBackend, "langchain-huggingface": LCHuggingFaceBackend, "transformers-musicgen": TransformersBackend, + "sentencetranformers": TransformersBackend, +} + +var TypeAlias map[string]string = map[string]string{ + "sentencetranformers": "SentenceTransformer", } var AutoDetect = os.Getenv("DISABLE_AUTODETECT") != "true" @@ -461,6 +466,10 @@ func (ml *ModelLoader) backendLoader(opts ...Option) (client grpc.Backend, err e backend := strings.ToLower(o.backendString) if realBackend, exists := Aliases[backend]; exists { backend = realBackend + typeAlias, exists := TypeAlias[backend] + if exists { + o.gRPCOptions.Type = typeAlias + } log.Debug().Msgf("%s is an alias of %s", backend, realBackend) }