From 0936086629f76d8029b4122a1c6a1a0978ed9f89 Mon Sep 17 00:00:00 2001 From: Ettore Di Giacinto Date: Fri, 17 Jan 2025 19:29:47 +0100 Subject: [PATCH] Add alias also for transformers-musicgen Signed-off-by: Ettore Di Giacinto --- pkg/model/initializers.go | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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"