diff --git a/pkg/model/initializers.go b/pkg/model/initializers.go index 0c964f2a..856e5463 100644 --- 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"