ExternalVendorCode/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2025-06-14 21:18:07 +00:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

feat(grpc): backend SPI pluggable in embedding mode (#1621)

Browse Source 
* run server

* grpc backend embedded support

* backend providable
This commit is contained in:
coyzeng
2024-01-23 15:56:36 +08:00
committed by GitHub
parent efe2883c5d
commit d5d82ba344
8 changed files with 196 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
api/backend/embeddings.go
View File

@ -41,7 +41,7 @@ func ModelEmbedding(s string, tokens []int, loader *model.ModelLoader, c config.
var fn func() ([]float32, error)
switch model := inferenceModel.(type) {
case *grpc.Client:
case grpc.Backend:
fn = func() ([]float32, error) {
predictOptions := gRPCPredictOpts(c, loader.ModelPath)
if len(tokens) > 0 {