LukeParke/LocalAI
1
0
Fork 0
mirror of https://github.com/LukeHagar/LocalAI.git synced 2025-12-09 12:37:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add image generation with ncnn-stablediffusion (#272)

Browse Source
This commit is contained in:
Ettore Di Giacinto
2023-05-16 19:32:53 +02:00
committed by GitHub
parent acd03d15f2
commit 9d051c5d4f
17 changed files with 582 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
api/api.go
View File

@@ -12,7 +12,7 @@ import (
"github.com/rs/zerolog/log"
)
func App(configFile string, loader *model.ModelLoader, uploadLimitMB, threads, ctxSize int, f16 bool, debug, disableMessage bool) *fiber.App {
func App(configFile string, loader *model.ModelLoader, uploadLimitMB, threads, ctxSize int, f16 bool, debug, disableMessage bool, imageDir string) *fiber.App {
zerolog.SetGlobalLevel(zerolog.InfoLevel)
if debug {
zerolog.SetGlobalLevel(zerolog.DebugLevel)
@@ -87,6 +87,12 @@ func App(configFile string, loader *model.ModelLoader, uploadLimitMB, threads, c
app.Post("/v1/audio/transcriptions", transcriptEndpoint(cm, debug, loader, threads, ctxSize, f16))
app.Post("/v1/images/generations", imageEndpoint(cm, debug, loader, imageDir))
if imageDir != "" {
app.Static("/generated-images", imageDir)
}
app.Get("/v1/models", listModels(loader, cm))
app.Get("/models", listModels(loader, cm))