ExternalVendorCode/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2025-06-15 21:48:07 +00:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

Feat: new backend: transformers-musicgen (#1387)

Browse Source 
Transformers-MusicGen
---------

Signed-off-by: Dave <dave@gray101.com>
This commit is contained in:
Dave
2023-12-08 04:01:02 -05:00
committed by GitHub
parent 6011911746
commit 8b6e601405
26 changed files with 868 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
api/backend/tts.go
View File

@ -59,9 +59,13 @@ func ModelTTS(backend, text, modelFile string, loader *model.ModelLoader, o *opt
// If the model file is not empty, we pass it joined with the model path
modelPath := ""
if modelFile != "" {
modelPath = filepath.Join(o.Loader.ModelPath, modelFile)
if err := utils.VerifyPath(modelPath, o.Loader.ModelPath); err != nil {
return "", nil, err
if bb != model.TransformersMusicGen {
modelPath = filepath.Join(o.Loader.ModelPath, modelFile)
if err := utils.VerifyPath(modelPath, o.Loader.ModelPath); err != nil {
return "", nil, err
}
} else {
modelPath = modelFile
}
}