UNPKG

@huggingface/inference

Version:

Typescript client for the Hugging Face Inference Providers and Inference Endpoints

37 lines 5.65 kB
import type { AudioClassificationTaskHelper, AudioToAudioTaskHelper, AutomaticSpeechRecognitionTaskHelper, ConversationalTaskHelper, DocumentQuestionAnsweringTaskHelper, FeatureExtractionTaskHelper, FillMaskTaskHelper, ImageClassificationTaskHelper, ImageSegmentationTaskHelper, ImageToImageTaskHelper, ImageToTextTaskHelper, ObjectDetectionTaskHelper, QuestionAnsweringTaskHelper, SentenceSimilarityTaskHelper, SummarizationTaskHelper, TableQuestionAnsweringTaskHelper, TabularClassificationTaskHelper, TabularRegressionTaskHelper, TaskProviderHelper, TextClassificationTaskHelper, TextGenerationTaskHelper, TextToAudioTaskHelper, TextToImageTaskHelper, TextToSpeechTaskHelper, TextToVideoTaskHelper, TokenClassificationTaskHelper, TranslationTaskHelper, VisualQuestionAnsweringTaskHelper, ZeroShotClassificationTaskHelper, ZeroShotImageClassificationTaskHelper } from "../providers/providerHelper"; import type { InferenceProvider, InferenceTask } from "../types"; export declare const PROVIDERS: Record<InferenceProvider, Partial<Record<InferenceTask, TaskProviderHelper>>>; /** * Get provider helper instance by name and task */ export declare function getProviderHelper(provider: InferenceProvider, task: "text-to-image"): TextToImageTaskHelper & TaskProviderHelper; export declare function getProviderHelper(provider: InferenceProvider, task: "conversational"): ConversationalTaskHelper & TaskProviderHelper; export declare function getProviderHelper(provider: InferenceProvider, task: "text-generation"): TextGenerationTaskHelper & TaskProviderHelper; export declare function getProviderHelper(provider: InferenceProvider, task: "text-to-speech"): TextToSpeechTaskHelper & TaskProviderHelper; export declare function getProviderHelper(provider: InferenceProvider, task: "text-to-audio"): TextToAudioTaskHelper & TaskProviderHelper; export declare function getProviderHelper(provider: InferenceProvider, task: "automatic-speech-recognition"): AutomaticSpeechRecognitionTaskHelper & TaskProviderHelper; export declare function getProviderHelper(provider: InferenceProvider, task: "text-to-video"): TextToVideoTaskHelper & TaskProviderHelper; export declare function getProviderHelper(provider: InferenceProvider, task: "text-classification"): TextClassificationTaskHelper & TaskProviderHelper; export declare function getProviderHelper(provider: InferenceProvider, task: "question-answering"): QuestionAnsweringTaskHelper & TaskProviderHelper; export declare function getProviderHelper(provider: InferenceProvider, task: "audio-classification"): AudioClassificationTaskHelper & TaskProviderHelper; export declare function getProviderHelper(provider: InferenceProvider, task: "audio-to-audio"): AudioToAudioTaskHelper & TaskProviderHelper; export declare function getProviderHelper(provider: InferenceProvider, task: "fill-mask"): FillMaskTaskHelper & TaskProviderHelper; export declare function getProviderHelper(provider: InferenceProvider, task: "feature-extraction"): FeatureExtractionTaskHelper & TaskProviderHelper; export declare function getProviderHelper(provider: InferenceProvider, task: "image-classification"): ImageClassificationTaskHelper & TaskProviderHelper; export declare function getProviderHelper(provider: InferenceProvider, task: "image-segmentation"): ImageSegmentationTaskHelper & TaskProviderHelper; export declare function getProviderHelper(provider: InferenceProvider, task: "document-question-answering"): DocumentQuestionAnsweringTaskHelper & TaskProviderHelper; export declare function getProviderHelper(provider: InferenceProvider, task: "image-to-text"): ImageToTextTaskHelper & TaskProviderHelper; export declare function getProviderHelper(provider: InferenceProvider, task: "object-detection"): ObjectDetectionTaskHelper & TaskProviderHelper; export declare function getProviderHelper(provider: InferenceProvider, task: "zero-shot-image-classification"): ZeroShotImageClassificationTaskHelper & TaskProviderHelper; export declare function getProviderHelper(provider: InferenceProvider, task: "zero-shot-classification"): ZeroShotClassificationTaskHelper & TaskProviderHelper; export declare function getProviderHelper(provider: InferenceProvider, task: "image-to-image"): ImageToImageTaskHelper & TaskProviderHelper; export declare function getProviderHelper(provider: InferenceProvider, task: "sentence-similarity"): SentenceSimilarityTaskHelper & TaskProviderHelper; export declare function getProviderHelper(provider: InferenceProvider, task: "table-question-answering"): TableQuestionAnsweringTaskHelper & TaskProviderHelper; export declare function getProviderHelper(provider: InferenceProvider, task: "tabular-classification"): TabularClassificationTaskHelper & TaskProviderHelper; export declare function getProviderHelper(provider: InferenceProvider, task: "tabular-regression"): TabularRegressionTaskHelper & TaskProviderHelper; export declare function getProviderHelper(provider: InferenceProvider, task: "token-classification"): TokenClassificationTaskHelper & TaskProviderHelper; export declare function getProviderHelper(provider: InferenceProvider, task: "translation"): TranslationTaskHelper & TaskProviderHelper; export declare function getProviderHelper(provider: InferenceProvider, task: "summarization"): SummarizationTaskHelper & TaskProviderHelper; export declare function getProviderHelper(provider: InferenceProvider, task: "visual-question-answering"): VisualQuestionAnsweringTaskHelper & TaskProviderHelper; export declare function getProviderHelper(provider: InferenceProvider, task: InferenceTask | undefined): TaskProviderHelper; //# sourceMappingURL=getProviderHelper.d.ts.map