dearwolf/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2025-05-20 02:24:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 21

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
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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())