dearwolf/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2025-06-29 22:20:43 +00:00
Code Issues Projects Releases Packages Wiki Activity

Add alias to silently redirect sentencetransformers to transformers

Browse source 
Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
This commit is contained in:
Ettore Di Giacinto 2025-01-17 19:27:15 +01:00
parent 981310c94f
commit beccf0a336
1 changed files with 9 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

9
pkg/model/initializers.go
View file

@ -27,6 +27,11 @@ var Aliases map[string]string = map[string]string{
"embedded-store": LocalStoreBackend, "embedded-store": LocalStoreBackend,
"langchain-huggingface": LCHuggingFaceBackend, "langchain-huggingface": LCHuggingFaceBackend,
"transformers-musicgen": TransformersBackend, "transformers-musicgen": TransformersBackend,
"sentencetranformers": TransformersBackend,
}
var TypeAlias map[string]string = map[string]string{
"sentencetranformers": "SentenceTransformer",
} }
var AutoDetect = os.Getenv("DISABLE_AUTODETECT") != "true" 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) backend := strings.ToLower(o.backendString)
if realBackend, exists := Aliases[backend]; exists { if realBackend, exists := Aliases[backend]; exists {
backend = realBackend backend = realBackend
typeAlias, exists := TypeAlias[backend]
if exists {
o.gRPCOptions.Type = typeAlias
}
log.Debug().Msgf("%s is an alias of %s", backend, realBackend) log.Debug().Msgf("%s is an alias of %s", backend, realBackend)
} }