chore: alias transformers-musicgen to transformers (#4623)

chore: alias transformers-muscigen to transformers Signed-off-by: Ettore Di Giacinto <[email protected]>
mudler · Jan 17, 2025 · a761e01 · a761e01
1 parent 96f8ec0
commit a761e01
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pkg/model/initializers.go b/pkg/model/initializers.go
@@ -26,6 +26,7 @@ var Aliases map[string]string = map[string]string{
 	"llama":                 LLamaCPP,
 	"embedded-store":        LocalStoreBackend,
 	"langchain-huggingface": LCHuggingFaceBackend,
+	"transformers-musicgen": TransformersBackend,
 }
 
 var AutoDetect = os.Getenv("DISABLE_AUTODETECT") != "true"
@@ -51,7 +52,8 @@ const (
 	PiperBackend           = "piper"
 	LCHuggingFaceBackend   = "huggingface"
 
-	LocalStoreBackend = "local-store"
+	TransformersBackend = "transformers"
+	LocalStoreBackend   = "local-store"
 )
 
 func backendPath(assetDir, backend string) string {