mirror/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2026-01-05 10:10:08 -06:00
Code Issues Packages Projects Releases Wiki Activity

feat: add tiny dream stable diffusion support (#1283)

Browse Source 
Signed-off-by: Gianluca Boiano <morf3089@gmail.com>
This commit is contained in:
Gianluca Boiano
2023-12-24 20:27:24 +01:00
committed by GitHub
parent f7621b2c6c
commit cae7b197ec
13 changed files with 161 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
api/openai/image.go
View File

@@ -122,8 +122,12 @@ func ImageEndpoint(cm *config.ConfigLoader, o *options.Option) func(c *fiber.Ctx
log.Debug().Msgf("Parameter Config: %+v", config)
// XXX: Only stablediffusion is supported for now
if config.Backend == "" {
switch config.Backend {
case "stablediffusion":
config.Backend = model.StableDiffusionBackend
case "tinydream":
config.Backend = model.TinyDreamBackend
default:
config.Backend = model.StableDiffusionBackend
}