Spaces:
Running
Running
| export const Task = Object.freeze({ | |
| TEXT_CLASSIFICATION: "text-classification", | |
| TOKEN_CLASSIFICATION: "token-classification", | |
| TABLE_QUESTION_ANSWERING: "table-question-answering", | |
| QUESTION_ANSWERING: "question-answering", | |
| ZERO_SHOT_CLASSIFICATION: "zero-shot-classification", | |
| TRANSLATION: "translation", | |
| SUMMARIZATION: "summarization", | |
| CONVERSATIONAL: "conversational", | |
| FEATURE_EXTRACTION: "feature-extraction", | |
| TEXT_GENERATION: "text-generation", | |
| TEXT2TEXT_GENERATION: "text2text-generation", | |
| FILL_MASK: "fill-mask", | |
| SENTENCE_SIMILARITY: "sentence-similarity", | |
| TEXT_TO_SPEECH: "text-to-speech", | |
| AUTOMATIC_SPEECH_RECOGNITION: "automatic-speech-recognition", | |
| AUDIO_TO_AUDIO: "audio-to-audio", | |
| AUDIO_CLASSIFICATION: "audio-classification", | |
| VOICE_ACTIVITY_DETECTION: "voice-activity-detection", | |
| DEPTH_ESTIMATION: "depth-estimation", | |
| IMAGE_CLASSIFICATION: "image-classification", | |
| OBJECT_DETECTION: "object-detection", | |
| IMAGE_SEGMENTATION: "image-segmentation", | |
| TEXT_TO_IMAGE: "text-to-image", | |
| IMAGE_TO_TEXT: "image-to-text", | |
| IMAGE_TO_IMAGE: "image-to-image", | |
| UNCONDITIONAL_IMAGE_GENERATION: "unconditional-image-generation", | |
| VIDEO_CLASSIFICATION: "video-classification", | |
| REINFORCEMENT_LEARNING: "reinforcement-learning", | |
| ROBOTICS: "robotics", | |
| TABULAR_CLASSIFICATION: "tabular-classification", | |
| TABULAR_REGRESSION: "tabular-regression", | |
| TABULAR_TO_TEXT: "tabular-to-text", | |
| TABLE_TO_TEXT: "table-to-text", | |
| MULTIPLE_CHOICE: "multiple-choice", | |
| TEXT_RETRIEVAL: "text-retrieval", | |
| TIME_SERIES_FORECASTING: "time-series-forecasting", | |
| VISUAL_QUESTION_ANSWERING: "visual-question-answering", | |
| DOCUMENT_QUESTION_ANSWERING: "document-question-answering", | |
| ZERO_SHOT_IMAGE_CLASSIFICATION: "zero-shot-image-classification", | |
| GRAPH_ML: "graph-ml", | |
| OTHER: "other", | |
| }); | |
| export const textGenerationModels = [ | |
| "bigcode/starcoder", | |
| "meta-llama/Llama-2-7b-chat-hf", | |
| "meta-llama/Llama-2-70b-chat-hf", | |
| "tiiuae/falcon-7b", | |
| "meta-llama/Llama-2-7b-hf", | |
| "tiiuae/falcon-7b-instruct", | |
| "anon8231489123/vicuna-13b-GPTQ-4bit-128g", | |
| "meta-llama/Llama-2-13b-chat-hf", | |
| "meta-llama/Llama-2-70b-hf", | |
| "Gustavosta/MagicPrompt-Stable-Diffusion" | |
| ] | |