LocalAI / core /backend /stores.go
AbdulElahGwaith's picture
Upload folder using huggingface_hub
0f07ba7 verified
package backend
import (
"github.com/mudler/LocalAI/core/config"
"github.com/mudler/LocalAI/pkg/grpc"
"github.com/mudler/LocalAI/pkg/model"
)
func StoreBackend(sl *model.ModelLoader, appConfig *config.ApplicationConfig, storeName string, backend string) (grpc.Backend, error) {
if backend == "" {
backend = model.LocalStoreBackend
}
sc := []model.Option{
model.WithBackendString(backend),
model.WithModel(storeName),
}
return sl.Load(sc...)
}