From 7dcc9c791edddcd4391b390511ed4a1ec6d7c6fc Mon Sep 17 00:00:00 2001
From: RayanMarmar <106345540+RayanMarmar@users.noreply.github.com>
Date: Wed, 15 May 2024 10:37:25 +0200
Subject: [PATCH] #158 added more compatible modules (#159)

---
 pkg/huggingface/tag.go | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/pkg/huggingface/tag.go b/pkg/huggingface/tag.go
index 2c52a88e..c279abd3 100644
--- a/pkg/huggingface/tag.go
+++ b/pkg/huggingface/tag.go
@@ -3,6 +3,11 @@ package huggingface
 var AllTags = []PipelineTag{
 	TextGeneration,
 	TextToImage,
+	TextToVideo,
+	TextToAudio,
+	TextTo3d,
+	ImageToText,
+	ImageToImage,
 }
 
 type PipelineTag string
@@ -10,6 +15,11 @@ 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