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

feat: allow to preload models before startup via env var or configs (#391)

Browse Source
This commit is contained in:
Ettore Di Giacinto
2023-05-27 09:26:33 +02:00
committed by GitHub
parent 835a20610b
commit 76c881043e
4 changed files with 112 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
api/api.go
View File

@ -69,6 +69,18 @@ func App(opts ...AppOption) *fiber.App {
// Default middleware config
app.Use(recover.New())
if options.preloadJSONModels != "" {
if err := ApplyGalleryFromString(options.loader.ModelPath, options.preloadJSONModels, cm); err != nil {
return nil
}
}
if options.preloadModelsFromPath != "" {
if err := ApplyGalleryFromFile(options.loader.ModelPath, options.preloadModelsFromPath, cm); err != nil {
return nil
}
}
if options.cors {
if options.corsAllowOrigins == "" {
app.Use(cors.New())