Spaces:
Paused
Paused
| ; | |
| Object.defineProperty(exports, "__esModule", { value: true }); | |
| exports.REMOVED_IN_V5_TRANSFORMERS_PIPELINES = exports.LIBRARY_TASK_MAPPING = void 0; | |
| /** | |
| * Mapping from library name to its supported tasks. | |
| * HF-Inference API (serverless) should be disabled for all other (library, task) pairs beyond this mapping. | |
| * This mapping is partially generated automatically by "python-api-export-tasks" action in | |
| * huggingface/api-inference-community repo upon merge. For transformers, the mapping is manually | |
| * based on api-inference (hf_types.rs). | |
| */ | |
| exports.LIBRARY_TASK_MAPPING = { | |
| "adapter-transformers": ["question-answering", "text-classification", "token-classification"], | |
| allennlp: ["question-answering"], | |
| asteroid: [ | |
| // "audio-source-separation", | |
| "audio-to-audio", | |
| ], | |
| bertopic: ["text-classification"], | |
| diffusers: ["image-to-image", "text-to-image"], | |
| doctr: ["object-detection"], | |
| espnet: ["text-to-speech", "automatic-speech-recognition"], | |
| fairseq: ["text-to-speech", "audio-to-audio"], | |
| fastai: ["image-classification"], | |
| fasttext: ["feature-extraction", "text-classification"], | |
| flair: ["token-classification"], | |
| k2: ["automatic-speech-recognition"], | |
| keras: ["image-classification"], | |
| nemo: ["automatic-speech-recognition"], | |
| open_clip: ["zero-shot-classification", "zero-shot-image-classification"], | |
| paddlenlp: ["fill-mask", "summarization", "zero-shot-classification"], | |
| peft: ["text-generation"], | |
| "pyannote-audio": ["automatic-speech-recognition"], | |
| "sentence-transformers": ["feature-extraction", "sentence-similarity"], | |
| setfit: ["text-classification"], | |
| sklearn: ["tabular-classification", "tabular-regression", "text-classification"], | |
| spacy: ["token-classification", "text-classification", "sentence-similarity"], | |
| "span-marker": ["token-classification"], | |
| speechbrain: ["audio-classification", "audio-to-audio", "automatic-speech-recognition", "text-to-speech"], | |
| stanza: ["token-classification"], | |
| timm: ["image-classification", "image-feature-extraction"], | |
| transformers: [ | |
| "audio-classification", | |
| "automatic-speech-recognition", | |
| "depth-estimation", | |
| "document-question-answering", | |
| "feature-extraction", | |
| "fill-mask", | |
| "image-classification", | |
| "image-feature-extraction", | |
| "image-segmentation", | |
| "image-to-image", | |
| "image-to-text", | |
| "image-text-to-text", | |
| "mask-generation", | |
| "object-detection", | |
| "question-answering", | |
| "summarization", | |
| "table-question-answering", | |
| "text-classification", | |
| "text-generation", | |
| "text-to-audio", | |
| "text-to-speech", | |
| "token-classification", | |
| "translation", | |
| "video-classification", | |
| "visual-question-answering", | |
| "zero-shot-classification", | |
| "zero-shot-image-classification", | |
| "zero-shot-object-detection", | |
| ], | |
| mindspore: ["image-classification"], | |
| }; | |
| // Pipeline types that were supported in legacy transformers versions (<5.0.0) | |
| exports.REMOVED_IN_V5_TRANSFORMERS_PIPELINES = ["image-to-text", "summarization", "translation"]; | |