Exrends list of supported model types for the MMD

Author: mrDzurb

ads/aqua/model/enums.py

@@ -27,6 +27,24 @@ class FineTuningCustomMetadata(ExtendedEnum):
 
 class MultiModelSupportedTaskType(ExtendedEnum):
     TEXT_GENERATION = "text_generation"
+    TEXT_GENERATION_INFERENCE = "text_generation_inference"
+    TEXT2TEXT_GENERATION = "text2text_generation"
+    SUMMARIZATION = "summarization"
+    TRANSLATION = "translation"
+    CONVERSATIONAL = "conversational"
+    FEATURE_EXTRACTION = "feature_extraction"
+    SENTENCE_SIMILARITY = "sentence_similarity"
+    AUTOMATIC_SPEECH_RECOGNITION = "automatic_speech_recognition"
+    TEXT_TO_SPEECH = "text_to_speech"
+    TEXT_TO_IMAGE = "text_to_image"
+    TEXT_EMBEDDING = "text_embedding"
     IMAGE_TEXT_TO_TEXT = "image_text_to_text"
     CODE_SYNTHESIS = "code_synthesis"
-    EMBEDDING = "text_embedding"
+    QUESTION_ANSWERING = "question_answering"
+    AUDIO_CLASSIFICATION = "audio_classification"
+    AUDIO_TO_AUDIO = "audio_to_audio"
+    IMAGE_CLASSIFICATION = "image_classification"
+    IMAGE_TO_TEXT = "image_to_text"
+    IMAGE_TO_IMAGE = "image_to_image"
+    VIDEO_CLASSIFICATION = "video_classification"
+    TIME_SERIES_FORECASTING = "time_series_forecasting"