@huggingface/inference
Version:
Typescript client for the Hugging Face Inference Providers and Inference Endpoints
37 lines • 5.65 kB
TypeScript
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