Hub Python 庫文件
推理型別
並獲得增強的文件體驗
開始使用
推理型別
本頁面列出了 Hugging Face Hub 支援的每項任務的可用型別（例如資料類）。每項任務都使用 JSON Schema 指定，這些型別是根據這些 Schema 生成的，並根據 Python 要求進行了一些定製。訪問 @huggingface.js/tasks 以查詢每項任務的 JSON Schema。
該庫的這一部分仍在開發中，將在未來版本中得到改進。
audio_classification
class huggingface_hub.AudioClassificationInput
< source >( inputs: str parameters: typing.Optional[huggingface_hub.inference._generated.types.audio_classification.AudioClassificationParameters] = None )
音訊分類推理的輸入
音訊分類推理的輸出
class huggingface_hub.AudioClassificationParameters
< source >( function_to_apply: typing.Optional[ForwardRef('AudioClassificationOutputTransform')] = None top_k: typing.Optional[int] = None )
音訊分類的額外推理引數
audio_to_audio
音訊到音訊推理的輸入
class huggingface_hub.AudioToAudioOutputElement
< source >( blob: typing.Any content_type: str label: str )
音訊到音訊任務的推理輸出 一個生成的帶標籤的音訊檔案。
自動語音識別
class huggingface_hub.AutomaticSpeechRecognitionGenerationParameters
< source >( do_sample: typing.Optional[bool] = None early_stopping: typing.Union[bool, ForwardRef('AutomaticSpeechRecognitionEarlyStoppingEnum'), NoneType] = None epsilon_cutoff: typing.Optional[float] = None eta_cutoff: typing.Optional[float] = None max_length: typing.Optional[int] = None max_new_tokens: typing.Optional[int] = None min_length: typing.Optional[int] = None min_new_tokens: typing.Optional[int] = None num_beam_groups: typing.Optional[int] = None num_beams: typing.Optional[int] = None penalty_alpha: typing.Optional[float] = None temperature: typing.Optional[float] = None top_k: typing.Optional[int] = None top_p: typing.Optional[float] = None typical_p: typing.Optional[float] = None use_cache: typing.Optional[bool] = None )
文字生成過程的引數化
class huggingface_hub.AutomaticSpeechRecognitionInput
< source >( inputs: str parameters: typing.Optional[huggingface_hub.inference._generated.types.automatic_speech_recognition.AutomaticSpeechRecognitionParameters] = None )
自動語音識別推理的輸入
class huggingface_hub.AutomaticSpeechRecognitionOutput
< source >( text: str chunks: typing.Optional[typing.List[huggingface_hub.inference._generated.types.automatic_speech_recognition.AutomaticSpeechRecognitionOutputChunk]] = None )
自動語音識別任務的推理輸出
class huggingface_hub.AutomaticSpeechRecognitionOutputChunk
< source >( text: str timestamp: typing.List[float] )
class huggingface_hub.AutomaticSpeechRecognitionParameters
< source >( generation_parameters: typing.Optional[huggingface_hub.inference._generated.types.automatic_speech_recognition.AutomaticSpeechRecognitionGenerationParameters] = None return_timestamps: typing.Optional[bool] = None )
自動語音識別的額外推理引數
聊天補全
class huggingface_hub.ChatCompletionInput
< source >( messages: typing.List[huggingface_hub.inference._generated.types.chat_completion.ChatCompletionInputMessage] frequency_penalty: typing.Optional[float] = None logit_bias: typing.Optional[typing.List[float]] = None logprobs: typing.Optional[bool] = None max_tokens: typing.Optional[int] = None model: typing.Optional[str] = None n: typing.Optional[int] = None presence_penalty: typing.Optional[float] = None response_format: typing.Union[huggingface_hub.inference._generated.types.chat_completion.ChatCompletionInputResponseFormatText, huggingface_hub.inference._generated.types.chat_completion.ChatCompletionInputResponseFormatJSONSchema, huggingface_hub.inference._generated.types.chat_completion.ChatCompletionInputResponseFormatJSONObject, NoneType] = None seed: typing.Optional[int] = None stop: typing.Optional[typing.List[str]] = None stream: typing.Optional[bool] = None stream_options: typing.Optional[huggingface_hub.inference._generated.types.chat_completion.ChatCompletionInputStreamOptions] = None temperature: typing.Optional[float] = None tool_choice: typing.Union[huggingface_hub.inference._generated.types.chat_completion.ChatCompletionInputToolChoiceClass, ForwardRef('ChatCompletionInputToolChoiceEnum'), NoneType] = None tool_prompt: typing.Optional[str] = None tools: typing.Optional[typing.List[huggingface_hub.inference._generated.types.chat_completion.ChatCompletionInputTool]] = None top_logprobs: typing.Optional[int] = None top_p: typing.Optional[float] = None )
聊天補全輸入。從 TGI 規範自動生成。更多詳細資訊，請檢視 https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-tgi-import.ts。
class huggingface_hub.ChatCompletionInputFunctionDefinition
< source >( name: str parameters: typing.Any description: typing.Optional[str] = None )
class huggingface_hub.ChatCompletionInputJSONSchema
< source >( name: str description: typing.Optional[str] = None schema: typing.Optional[typing.Dict[str, object]] = None strict: typing.Optional[bool] = None )
class huggingface_hub.ChatCompletionInputMessage
< source >( role: str content: typing.Union[typing.List[huggingface_hub.inference._generated.types.chat_completion.ChatCompletionInputMessageChunk], str, NoneType] = None name: typing.Optional[str] = None tool_calls: typing.Optional[typing.List[huggingface_hub.inference._generated.types.chat_completion.ChatCompletionInputToolCall]] = None )
class huggingface_hub.ChatCompletionInputMessageChunk
< source >( type: ChatCompletionInputMessageChunkType image_url: typing.Optional[huggingface_hub.inference._generated.types.chat_completion.ChatCompletionInputURL] = None text: typing.Optional[str] = None )
class huggingface_hub.ChatCompletionInputResponseFormatJSONObject
< source >( type: typing.Literal['json_object'] )
class huggingface_hub.ChatCompletionInputResponseFormatJSONSchema
< source >( type: typing.Literal['json_schema'] json_schema: ChatCompletionInputJSONSchema )
class huggingface_hub.ChatCompletionInputResponseFormatText
< source >( type: typing.Literal['text'] )
class huggingface_hub.ChatCompletionInputStreamOptions
< source >( include_usage: typing.Optional[bool] = None )
class huggingface_hub.ChatCompletionInputTool
< source >( function: ChatCompletionInputFunctionDefinition type: str )
class huggingface_hub.ChatCompletionInputToolCall
< source >( function: ChatCompletionInputFunctionDefinition id: str type: str )
class huggingface_hub.ChatCompletionInputToolChoiceClass
< source >( function: ChatCompletionInputFunctionName )
class huggingface_hub.ChatCompletionOutput
< source >( choices: typing.List[huggingface_hub.inference._generated.types.chat_completion.ChatCompletionOutputComplete] created: int id: str model: str system_fingerprint: str usage: ChatCompletionOutputUsage )
聊天補全輸出。從 TGI 規範自動生成。更多詳細資訊，請檢視 https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-tgi-import.ts。
class huggingface_hub.ChatCompletionOutputComplete
< source >( finish_reason: str index: int message: ChatCompletionOutputMessage logprobs: typing.Optional[huggingface_hub.inference._generated.types.chat_completion.ChatCompletionOutputLogprobs] = None )
class huggingface_hub.ChatCompletionOutputFunctionDefinition
< source >( arguments: str name: str description: typing.Optional[str] = None )
class huggingface_hub.ChatCompletionOutputLogprob
< source >( logprob: float token: str top_logprobs: typing.List[huggingface_hub.inference._generated.types.chat_completion.ChatCompletionOutputTopLogprob] )
class huggingface_hub.ChatCompletionOutputLogprobs
< 來源 >( content: typing.List[huggingface_hub.inference._generated.types.chat_completion.ChatCompletionOutputLogprob] )
class huggingface_hub.ChatCompletionOutputMessage
< 來源 >( role: str content: typing.Optional[str] = None tool_call_id: typing.Optional[str] = None tool_calls: typing.Optional[typing.List[huggingface_hub.inference._generated.types.chat_completion.ChatCompletionOutputToolCall]] = None )
class huggingface_hub.ChatCompletionOutputToolCall
< 來源 >( function: ChatCompletionOutputFunctionDefinition id: str type: str )
class huggingface_hub.ChatCompletionOutputUsage
< 來源 >( completion_tokens: int prompt_tokens: int total_tokens: int )
class huggingface_hub.ChatCompletionStreamOutput
< 來源 >( choices: typing.List[huggingface_hub.inference._generated.types.chat_completion.ChatCompletionStreamOutputChoice] created: int id: str model: str system_fingerprint: str usage: typing.Optional[huggingface_hub.inference._generated.types.chat_completion.ChatCompletionStreamOutputUsage] = None )
聊天完成流輸出。從TGI規範自動生成。有關更多詳細資訊，請檢視https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-tgi-import.ts。
class huggingface_hub.ChatCompletionStreamOutputChoice
< 來源 >( delta: ChatCompletionStreamOutputDelta index: int finish_reason: typing.Optional[str] = None logprobs: typing.Optional[huggingface_hub.inference._generated.types.chat_completion.ChatCompletionStreamOutputLogprobs] = None )
class huggingface_hub.ChatCompletionStreamOutputDelta
< 來源 >( role: str content: typing.Optional[str] = None tool_call_id: typing.Optional[str] = None tool_calls: typing.Optional[typing.List[huggingface_hub.inference._generated.types.chat_completion.ChatCompletionStreamOutputDeltaToolCall]] = None )
class huggingface_hub.ChatCompletionStreamOutputDeltaToolCall
< 來源 >( function: ChatCompletionStreamOutputFunction id: str index: int type: str )
class huggingface_hub.ChatCompletionStreamOutputFunction
< 來源 >( arguments: str name: typing.Optional[str] = None )
class huggingface_hub.ChatCompletionStreamOutputLogprob
< 來源 >( logprob: float token: str top_logprobs: typing.List[huggingface_hub.inference._generated.types.chat_completion.ChatCompletionStreamOutputTopLogprob] )
class huggingface_hub.ChatCompletionStreamOutputLogprobs
< 來源 >( content: typing.List[huggingface_hub.inference._generated.types.chat_completion.ChatCompletionStreamOutputLogprob] )
class huggingface_hub.ChatCompletionStreamOutputUsage
< 來源 >( completion_tokens: int prompt_tokens: int total_tokens: int )
depth_estimation
class huggingface_hub.DepthEstimationInput
< 來源 >( inputs: typing.Any parameters: typing.Optional[typing.Dict[str, typing.Any]] = None )
深度估計推理的輸入
class huggingface_hub.DepthEstimationOutput
< 來源 >( depth: typing.Any predicted_depth: typing.Any )
深度估計任務的推理輸出
document_question_answering
class huggingface_hub.DocumentQuestionAnsweringInput
< 來源 >( inputs: DocumentQuestionAnsweringInputData parameters: typing.Optional[huggingface_hub.inference._generated.types.document_question_answering.DocumentQuestionAnsweringParameters] = None )
文件問答推理的輸入
一個（文件，問題）對以回答
class huggingface_hub.DocumentQuestionAnsweringOutputElement
< 來源 >( answer: str end: int score: float start: int )
文件問答任務的推理輸出
class huggingface_hub.DocumentQuestionAnsweringParameters
< 來源 >( doc_stride: typing.Optional[int] = None handle_impossible_answer: typing.Optional[bool] = None lang: typing.Optional[str] = None max_answer_len: typing.Optional[int] = None max_question_len: typing.Optional[int] = None max_seq_len: typing.Optional[int] = None top_k: typing.Optional[int] = None word_boxes: typing.Optional[typing.List[typing.Union[typing.List[float], str]]] = None )
文件問答的其他推理引數
feature_extraction
class huggingface_hub.FeatureExtractionInput
< 來源 >( inputs: typing.Union[typing.List[str], str] normalize: typing.Optional[bool] = None prompt_name: typing.Optional[str] = None truncate: typing.Optional[bool] = None truncation_direction: typing.Optional[ForwardRef('FeatureExtractionInputTruncationDirection')] = None )
特徵提取輸入。從TEI規範自動生成。有關更多詳細資訊，請檢視https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-tei-import.ts。
fill_mask
class huggingface_hub.FillMaskInput
< 來源 >( inputs: str parameters: typing.Optional[huggingface_hub.inference._generated.types.fill_mask.FillMaskParameters] = None )
填充掩碼推理的輸入
class huggingface_hub.FillMaskOutputElement
< 來源 >( score: float sequence: str token: int token_str: typing.Any fill_mask_output_token_str: typing.Optional[str] = None )
填充掩碼任務的推理輸出
class huggingface_hub.FillMaskParameters
< 來源 >( targets: typing.Optional[typing.List[str]] = None top_k: typing.Optional[int] = None )
填充掩碼的其他推理引數
image_classification
class huggingface_hub.ImageClassificationInput
< 來源 >( inputs: str parameters: typing.Optional[huggingface_hub.inference._generated.types.image_classification.ImageClassificationParameters] = None )
影像分類推理的輸入
影像分類任務的推理輸出
class huggingface_hub.ImageClassificationParameters
< 來源 >( function_to_apply: typing.Optional[ForwardRef('ImageClassificationOutputTransform')] = None top_k: typing.Optional[int] = None )
影像分類的其他推理引數
image_segmentation
class huggingface_hub.ImageSegmentationInput
< 來源 >( inputs: str parameters: typing.Optional[huggingface_hub.inference._generated.types.image_segmentation.ImageSegmentationParameters] = None )
影像分割推理的輸入
class huggingface_hub.ImageSegmentationOutputElement
< 來源 >( label: str mask: str score: typing.Optional[float] = None )
影像分割任務的推理輸出，預測的掩碼/片段
class huggingface_hub.ImageSegmentationParameters
< 來源 >( mask_threshold: typing.Optional[float] = None overlap_mask_area_threshold: typing.Optional[float] = None subtask: typing.Optional[ForwardRef('ImageSegmentationSubtask')] = None threshold: typing.Optional[float] = None )
影像分割的其他推理引數
image_to_image
class huggingface_hub.ImageToImageInput
< 來源 >( inputs: str parameters: typing.Optional[huggingface_hub.inference._generated.types.image_to_image.ImageToImageParameters] = None )
影像到影像推理的輸入
影像到影像任務的推理輸出
class huggingface_hub.ImageToImageParameters
< 來源 >( guidance_scale: typing.Optional[float] = None negative_prompt: typing.Optional[str] = None num_inference_steps: typing.Optional[int] = None prompt: typing.Optional[str] = None target_size: typing.Optional[huggingface_hub.inference._generated.types.image_to_image.ImageToImageTargetSize] = None )
影像到影像的其他推理引數
輸出影像的畫素大小。
image_to_text
class huggingface_hub.ImageToTextGenerationParameters
< 來源 >( do_sample: typing.Optional[bool] = None early_stopping: typing.Union[bool, ForwardRef('ImageToTextEarlyStoppingEnum'), NoneType] = None epsilon_cutoff: typing.Optional[float] = None eta_cutoff: typing.Optional[float] = None max_length: typing.Optional[int] = None max_new_tokens: typing.Optional[int] = None min_length: typing.Optional[int] = None min_new_tokens: typing.Optional[int] = None num_beam_groups: typing.Optional[int] = None num_beams: typing.Optional[int] = None penalty_alpha: typing.Optional[float] = None temperature: typing.Optional[float] = None top_k: typing.Optional[int] = None top_p: typing.Optional[float] = None typical_p: typing.Optional[float] = None use_cache: typing.Optional[bool] = None )
文字生成過程的引數化
class huggingface_hub.ImageToTextInput
< 來源 >( inputs: typing.Any parameters: typing.Optional[huggingface_hub.inference._generated.types.image_to_text.ImageToTextParameters] = None )
影像到文字推理的輸入
class huggingface_hub.ImageToTextOutput
< 源 >( generated_text: typing.Any image_to_text_output_generated_text: typing.Optional[str] = None )
影像轉文字任務的推理輸出
類 huggingface_hub.ImageToTextParameters
< 源 >( generation_parameters: typing.Optional[huggingface_hub.inference._generated.types.image_to_text.ImageToTextGenerationParameters] = None max_new_tokens: typing.Optional[int] = None )
影像轉文字的其他推理引數
image_to_video (影像轉影片)
類 huggingface_hub.ImageToVideoInput
< 源 >( inputs: str parameters: typing.Optional[huggingface_hub.inference._generated.types.image_to_video.ImageToVideoParameters] = None )
影像轉影片推理的輸入
影像轉影片任務的推理輸出
類 huggingface_hub.ImageToVideoParameters
< 源 >( guidance_scale: typing.Optional[float] = None negative_prompt: typing.Optional[str] = None num_frames: typing.Optional[float] = None num_inference_steps: typing.Optional[int] = None prompt: typing.Optional[str] = None seed: typing.Optional[int] = None target_size: typing.Optional[huggingface_hub.inference._generated.types.image_to_video.ImageToVideoTargetSize] = None )
影像轉影片的其他推理引數
輸出影片幀的畫素大小。
object_detection (目標檢測)
預測的邊界框。座標是相對於輸入影像左上角的。
類 huggingface_hub.ObjectDetectionInput
< 源 >( inputs: str parameters: typing.Optional[huggingface_hub.inference._generated.types.object_detection.ObjectDetectionParameters] = None )
目標檢測推理的輸入
類 huggingface_hub.ObjectDetectionOutputElement
< 源 >( box: ObjectDetectionBoundingBox label: str score: float )
目標檢測任務的推理輸出
目標檢測的其他推理引數
question_answering (問答)
類 huggingface_hub.QuestionAnsweringInput
< 源 >( inputs: QuestionAnsweringInputData parameters: typing.Optional[huggingface_hub.inference._generated.types.question_answering.QuestionAnsweringParameters] = None )
問答推理的輸入
一個 (上下文, 問題) 對用於回答
類 huggingface_hub.QuestionAnsweringOutputElement
< 源 >( answer: str end: int score: float start: int )
問答任務的推理輸出
類 huggingface_hub.QuestionAnsweringParameters
< 源 >( align_to_words: typing.Optional[bool] = None doc_stride: typing.Optional[int] = None handle_impossible_answer: typing.Optional[bool] = None max_answer_len: typing.Optional[int] = None max_question_len: typing.Optional[int] = None max_seq_len: typing.Optional[int] = None top_k: typing.Optional[int] = None )
問答的其他推理引數
sentence_similarity (句子相似度)
類 huggingface_hub.SentenceSimilarityInput
< 源 >( inputs: SentenceSimilarityInputData parameters: typing.Optional[typing.Dict[str, typing.Any]] = None )
句子相似度推理的輸入
類 huggingface_hub.SentenceSimilarityInputData
< 源 >( sentences: typing.List[str] source_sentence: str )
summarization (摘要)
類 huggingface_hub.SummarizationInput
< 源 >( inputs: str parameters: typing.Optional[huggingface_hub.inference._generated.types.summarization.SummarizationParameters] = None )
摘要推理的輸入
摘要任務的推理輸出
類 huggingface_hub.SummarizationParameters
< 源 >( clean_up_tokenization_spaces: typing.Optional[bool] = None generate_parameters: typing.Optional[typing.Dict[str, typing.Any]] = None truncation: typing.Optional[ForwardRef('SummarizationTruncationStrategy')] = None )
摘要的其他推理引數。
table_question_answering (表格問答)
類 huggingface_hub.TableQuestionAnsweringInput
< 源 >( inputs: TableQuestionAnsweringInputData parameters: typing.Optional[huggingface_hub.inference._generated.types.table_question_answering.TableQuestionAnsweringParameters] = None )
表格問答推理的輸入
類 huggingface_hub.TableQuestionAnsweringInputData
< 源 >( question: str table: typing.Dict[str, typing.List[str]] )
一個 (表格, 問題) 對用於回答
類 huggingface_hub.TableQuestionAnsweringOutputElement
< 源 >( answer: str cells: typing.List[str] coordinates: typing.List[typing.List[int]] aggregator: typing.Optional[str] = None )
表格問答任務的推理輸出
類 huggingface_hub.TableQuestionAnsweringParameters
< 源 >( padding: typing.Optional[ForwardRef('Padding')] = None sequential: typing.Optional[bool] = None truncation: typing.Optional[bool] = None )
表格問答的其他推理引數
text2text_generation (文字到文字生成)
類 huggingface_hub.Text2TextGenerationInput
< 源 >( inputs: str parameters: typing.Optional[huggingface_hub.inference._generated.types.text2text_generation.Text2TextGenerationParameters] = None )
文字到文字生成推理的輸入
類 huggingface_hub.Text2TextGenerationOutput
< 源 >( generated_text: typing.Any text2_text_generation_output_generated_text: typing.Optional[str] = None )
文字到文字生成任務的推理輸出
類 huggingface_hub.Text2TextGenerationParameters
< 源 >( clean_up_tokenization_spaces: typing.Optional[bool] = None generate_parameters: typing.Optional[typing.Dict[str, typing.Any]] = None truncation: typing.Optional[ForwardRef('Text2TextGenerationTruncationStrategy')] = None )
文字到文字生成的其他推理引數
text_classification (文字分類)
類 huggingface_hub.TextClassificationInput
< 源 >( inputs: str parameters: typing.Optional[huggingface_hub.inference._generated.types.text_classification.TextClassificationParameters] = None )
文字分類推理的輸入
文字分類任務的推理輸出
類 huggingface_hub.TextClassificationParameters
< 源 >( function_to_apply: typing.Optional[ForwardRef('TextClassificationOutputTransform')] = None top_k: typing.Optional[int] = None )
文字分類的其他推理引數
text_generation (文字生成)
類 huggingface_hub.TextGenerationInput
< 源 >( inputs: str parameters: typing.Optional[huggingface_hub.inference._generated.types.text_generation.TextGenerationInputGenerateParameters] = None stream: typing.Optional[bool] = None )
文字生成輸入。從 TGI 規範自動生成。有關更多詳細資訊，請檢視 https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-tgi-import.ts。
類 huggingface_hub.TextGenerationInputGenerateParameters
< 源 >( adapter_id: typing.Optional[str] = None best_of: typing.Optional[int] = None decoder_input_details: typing.Optional[bool] = None details: typing.Optional[bool] = None do_sample: typing.Optional[bool] = None frequency_penalty: typing.Optional[float] = None grammar: typing.Optional[huggingface_hub.inference._generated.types.text_generation.TextGenerationInputGrammarType] = None max_new_tokens: typing.Optional[int] = None repetition_penalty: typing.Optional[float] = None return_full_text: typing.Optional[bool] = None seed: typing.Optional[int] = None stop: typing.Optional[typing.List[str]] = None temperature: typing.Optional[float] = None top_k: typing.Optional[int] = None top_n_tokens: typing.Optional[int] = None top_p: typing.Optional[float] = None truncate: typing.Optional[int] = None typical_p: typing.Optional[float] = None watermark: typing.Optional[bool] = None )
class huggingface_hub.TextGenerationOutput
< 源 >( generated_text: str details: typing.Optional[huggingface_hub.inference._generated.types.text_generation.TextGenerationOutputDetails] = None )
文字生成輸出。根據TGI規範自動生成。更多詳細資訊，請檢視 https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-tgi-import.ts。
class huggingface_hub.TextGenerationOutputBestOfSequence
< 源 >( finish_reason: TextGenerationOutputFinishReason generated_text: str generated_tokens: int prefill: typing.List[huggingface_hub.inference._generated.types.text_generation.TextGenerationOutputPrefillToken] tokens: typing.List[huggingface_hub.inference._generated.types.text_generation.TextGenerationOutputToken] seed: typing.Optional[int] = None top_tokens: typing.Optional[typing.List[typing.List[huggingface_hub.inference._generated.types.text_generation.TextGenerationOutputToken]]] = None )
class huggingface_hub.TextGenerationOutputDetails
< 源 >( finish_reason: TextGenerationOutputFinishReason generated_tokens: int prefill: typing.List[huggingface_hub.inference._generated.types.text_generation.TextGenerationOutputPrefillToken] tokens: typing.List[huggingface_hub.inference._generated.types.text_generation.TextGenerationOutputToken] best_of_sequences: typing.Optional[typing.List[huggingface_hub.inference._generated.types.text_generation.TextGenerationOutputBestOfSequence]] = None seed: typing.Optional[int] = None top_tokens: typing.Optional[typing.List[typing.List[huggingface_hub.inference._generated.types.text_generation.TextGenerationOutputToken]]] = None )
class huggingface_hub.TextGenerationOutputToken
< 源 >( id: int logprob: float special: bool text: str )
class huggingface_hub.TextGenerationStreamOutput
< 源 >( index: int token: TextGenerationStreamOutputToken details: typing.Optional[huggingface_hub.inference._generated.types.text_generation.TextGenerationStreamOutputStreamDetails] = None generated_text: typing.Optional[str] = None top_tokens: typing.Optional[typing.List[huggingface_hub.inference._generated.types.text_generation.TextGenerationStreamOutputToken]] = None )
文字生成流輸出。根據TGI規範自動生成。更多詳細資訊，請檢視 https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-tgi-import.ts。
class huggingface_hub.TextGenerationStreamOutputStreamDetails
< 源 >( finish_reason: TextGenerationOutputFinishReason generated_tokens: int input_length: int seed: typing.Optional[int] = None )
class huggingface_hub.TextGenerationStreamOutputToken
< 源 >( id: int logprob: float special: bool text: str )
text_to_audio
class huggingface_hub.TextToAudioGenerationParameters
< 源 >( do_sample: typing.Optional[bool] = None early_stopping: typing.Union[bool, ForwardRef('TextToAudioEarlyStoppingEnum'), NoneType] = None epsilon_cutoff: typing.Optional[float] = None eta_cutoff: typing.Optional[float] = None max_length: typing.Optional[int] = None max_new_tokens: typing.Optional[int] = None min_length: typing.Optional[int] = None min_new_tokens: typing.Optional[int] = None num_beam_groups: typing.Optional[int] = None num_beams: typing.Optional[int] = None penalty_alpha: typing.Optional[float] = None temperature: typing.Optional[float] = None top_k: typing.Optional[int] = None top_p: typing.Optional[float] = None typical_p: typing.Optional[float] = None use_cache: typing.Optional[bool] = None )
文字生成過程的引數化
class huggingface_hub.TextToAudioInput
< 源 >( inputs: str parameters: typing.Optional[huggingface_hub.inference._generated.types.text_to_audio.TextToAudioParameters] = None )
文字到音訊推理的輸入
文字到音訊任務的推理輸出
class huggingface_hub.TextToAudioParameters
< 源 >( generation_parameters: typing.Optional[huggingface_hub.inference._generated.types.text_to_audio.TextToAudioGenerationParameters] = None )
文字到音訊的其他推理引數
text_to_image
class huggingface_hub.TextToImageInput
< 源 >( inputs: str parameters: typing.Optional[huggingface_hub.inference._generated.types.text_to_image.TextToImageParameters] = None )
文字到影像推理的輸入
文字到影像任務的推理輸出
class huggingface_hub.TextToImageParameters
< 源 >( guidance_scale: typing.Optional[float] = None height: typing.Optional[int] = None negative_prompt: typing.Optional[str] = None num_inference_steps: typing.Optional[int] = None scheduler: typing.Optional[str] = None seed: typing.Optional[int] = None width: typing.Optional[int] = None )
文字到影像的其他推理引數
text_to_speech
class huggingface_hub.TextToSpeechGenerationParameters
< 源 >( do_sample: typing.Optional[bool] = None early_stopping: typing.Union[bool, ForwardRef('TextToSpeechEarlyStoppingEnum'), NoneType] = None epsilon_cutoff: typing.Optional[float] = None eta_cutoff: typing.Optional[float] = None max_length: typing.Optional[int] = None max_new_tokens: typing.Optional[int] = None min_length: typing.Optional[int] = None min_new_tokens: typing.Optional[int] = None num_beam_groups: typing.Optional[int] = None num_beams: typing.Optional[int] = None penalty_alpha: typing.Optional[float] = None temperature: typing.Optional[float] = None top_k: typing.Optional[int] = None top_p: typing.Optional[float] = None typical_p: typing.Optional[float] = None use_cache: typing.Optional[bool] = None )
文字生成過程的引數化
class huggingface_hub.TextToSpeechInput
< 源 >( inputs: str parameters: typing.Optional[huggingface_hub.inference._generated.types.text_to_speech.TextToSpeechParameters] = None )
文字到語音推理的輸入
class huggingface_hub.TextToSpeechOutput
< 源 >( audio: typing.Any sampling_rate: typing.Optional[float] = None )
文字到語音任務的推理輸出
class huggingface_hub.TextToSpeechParameters
< 源 >( generation_parameters: typing.Optional[huggingface_hub.inference._generated.types.text_to_speech.TextToSpeechGenerationParameters] = None )
文字到語音的其他推理引數
text_to_video
class huggingface_hub.TextToVideoInput
< 源 >( inputs: str parameters: typing.Optional[huggingface_hub.inference._generated.types.text_to_video.TextToVideoParameters] = None )
文字到影片推理的輸入
文字到影片任務的推理輸出
class huggingface_hub.TextToVideoParameters
< 源 >( guidance_scale: typing.Optional[float] = None negative_prompt: typing.Optional[typing.List[str]] = None num_frames: typing.Optional[float] = None num_inference_steps: typing.Optional[int] = None seed: typing.Optional[int] = None )
文字到影片的其他推理引數
token_classification
class huggingface_hub.TokenClassificationInput
< 源 >( inputs: str parameters: typing.Optional[huggingface_hub.inference._generated.types.token_classification.TokenClassificationParameters] = None )
令牌分類推理的輸入
class huggingface_hub.TokenClassificationOutputElement
< 源 >( end: int score: float start: int word: str entity: typing.Optional[str] = None entity_group: typing.Optional[str] = None )
令牌分類任務的推理輸出
class huggingface_hub.TokenClassificationParameters
< 源 >( aggregation_strategy: typing.Optional[ForwardRef('TokenClassificationAggregationStrategy')] = None ignore_labels: typing.Optional[typing.List[str]] = None stride: typing.Optional[int] = None )
令牌分類的其他推理引數
translation
class huggingface_hub.TranslationInput
< 源 >( inputs: str parameters: typing.Optional[huggingface_hub.inference._generated.types.translation.TranslationParameters] = None )
翻譯推理的輸入
翻譯任務的推理輸出
class huggingface_hub.TranslationParameters
< 源 >( clean_up_tokenization_spaces: typing.Optional[bool] = None generate_parameters: typing.Optional[typing.Dict[str, typing.Any]] = None src_lang: typing.Optional[str] = None tgt_lang: typing.Optional[str] = None truncation: typing.Optional[ForwardRef('TranslationTruncationStrategy')] = None )
翻譯的其他推理引數
video_classification
class huggingface_hub.VideoClassificationInput
< 源 >( inputs: typing.Any parameters: typing.Optional[huggingface_hub.inference._generated.types.video_classification.VideoClassificationParameters] = None )
影片分類推理的輸入
影片分類任務的推理輸出
class huggingface_hub.VideoClassificationParameters
< 源 >( frame_sampling_rate: typing.Optional[int] = None function_to_apply: typing.Optional[ForwardRef('VideoClassificationOutputTransform')] = None num_frames: typing.Optional[int] = None top_k: typing.Optional[int] = None )
影片分類的其他推理引數
視覺問答
class huggingface_hub.VisualQuestionAnsweringInput
< 源 >( inputs: VisualQuestionAnsweringInputData parameters: typing.Optional[huggingface_hub.inference._generated.types.visual_question_answering.VisualQuestionAnsweringParameters] = None )
視覺問答推理的輸入
一個 (影像, 問題) 對用於回答
class huggingface_hub.VisualQuestionAnsweringOutputElement
< 源 >( score: float answer: typing.Optional[str] = None )
視覺問答任務的推理輸出
class huggingface_hub.VisualQuestionAnsweringParameters
< 源 >( top_k: typing.Optional[int] = None )
視覺問答的其他推理引數
零樣本分類
class huggingface_hub.ZeroShotClassificationInput
< 源 >( inputs: str parameters: ZeroShotClassificationParameters )
零樣本分類推理的輸入
零樣本分類任務的推理輸出
class huggingface_hub.ZeroShotClassificationParameters
< 源 >( candidate_labels: typing.List[str] hypothesis_template: typing.Optional[str] = None multi_label: typing.Optional[bool] = None )
零樣本分類的其他推理引數
零樣本影像分類
class huggingface_hub.ZeroShotImageClassificationInput
< 源 >( inputs: str parameters: ZeroShotImageClassificationParameters )
零樣本影像分類推理的輸入
零樣本影像分類任務的推理輸出
class huggingface_hub.ZeroShotImageClassificationParameters
< 源 >( candidate_labels: typing.List[str] hypothesis_template: typing.Optional[str] = None )
零樣本影像分類的其他推理引數
零樣本目標檢測
class huggingface_hub.ZeroShotObjectDetectionBoundingBox
< 源 >( xmax: int xmin: int ymax: int ymin: int )
預測的邊界框。座標是相對於輸入影像左上角的。
class huggingface_hub.ZeroShotObjectDetectionInput
< 源 >( inputs: str parameters: ZeroShotObjectDetectionParameters )
零樣本目標檢測推理的輸入
class huggingface_hub.ZeroShotObjectDetectionOutputElement
< 源 >( box: ZeroShotObjectDetectionBoundingBox label: str score: float )
零樣本目標檢測任務的推理輸出
class huggingface_hub.ZeroShotObjectDetectionParameters
< 源 >( candidate_labels: typing.List[str] )
零樣本目標檢測的其他推理引數