#158 added more compatible modules (#159)

easy-model-fusion · May 15, 2024 · 7dcc9c7 · 7dcc9c7
1 parent 43f65fd
commit 7dcc9c7
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/pkg/huggingface/tag.go b/pkg/huggingface/tag.go
@@ -3,13 +3,23 @@ package huggingface
 var AllTags = []PipelineTag{
 	TextGeneration,
 	TextToImage,
+	TextToVideo,
+	TextToAudio,
+	TextTo3d,
+	ImageToText,
+	ImageToImage,
 }
 
 type PipelineTag string
 
 const (
 	TextGeneration PipelineTag = "text-generation"
 	TextToImage    PipelineTag = "text-to-image"
+	TextToVideo    PipelineTag = "text-to-video"
+	TextTo3d       PipelineTag = "text-to-3d"
+	TextToAudio    PipelineTag = "text-to-audio"
+	ImageToText    PipelineTag = "image-to-text"
+	ImageToImage   PipelineTag = "image-to-image"
 )
 
 // AllTagsString returns all tags as a string slice