add bloomz shortcut

batmac · Apr 21, 2023 · bf0f776 · bf0f776
1 parent ecbdc0a
commit bf0f776
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pkg/mutators/single/huggingface.go b/pkg/mutators/single/huggingface.go
@@ -51,6 +51,7 @@ var HuggingFaceCommonTasks = map[string]string{
 	"ner":             "dbmdz/bert-large-cased-finetuned-conll03-english",
 	"translation":     "t5-base",
 	"bloom":           "bigscience/bloom",
+	"bloomz":          "bigscience/bloomz",
 }
 
 type HuggingFaceRequest struct {