Add alias also for transformers-musicgen

Signed-off-by: Ettore Di Giacinto <[email protected]>
mudler · Jan 17, 2025 · 8ee1cfd · 8ee1cfd
1 parent 2641622
commit 8ee1cfd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pkg/model/initializers.go b/pkg/model/initializers.go
@@ -31,7 +31,8 @@ var Aliases map[string]string = map[string]string{
 }
 
 var TypeAlias map[string]string = map[string]string{
-	"sentencetranformers": "SentenceTransformer",
+	"sentencetranformers":   "SentenceTransformer",
+	"transformers-musicgen": "MusicgenForConditionalGeneration",
 }
 
 var AutoDetect = os.Getenv("DISABLE_AUTODETECT") != "true"