huggingface-hub 0.26.3__py3-none-any.whl → 0.27.0rc0__py3-none-any.whl

huggingface_hub/inference/_generated/types/__init__.py

@@ -24,10 +24,14 @@ from .chat_completion import (
     ChatCompletionInputFunctionDefinition,
     ChatCompletionInputFunctionName,
     ChatCompletionInputGrammarType,
+    ChatCompletionInputGrammarTypeType,
     ChatCompletionInputMessage,
     ChatCompletionInputMessageChunk,
+    ChatCompletionInputMessageChunkType,
     ChatCompletionInputStreamOptions,
-    ChatCompletionInputToolType,
+    ChatCompletionInputTool,
+    ChatCompletionInputToolChoiceClass,
+    ChatCompletionInputToolChoiceEnum,
     ChatCompletionInputURL,
     ChatCompletionOutput,
     ChatCompletionOutputComplete,
@@ -47,7 +51,6 @@ from .chat_completion import (
     ChatCompletionStreamOutputLogprobs,
     ChatCompletionStreamOutputTopLogprob,
     ChatCompletionStreamOutputUsage,
-    ToolElement,
 )
 from .depth_estimation import DepthEstimationInput, DepthEstimationOutput
 from .document_question_answering import (
@@ -56,7 +59,7 @@ from .document_question_answering import (
     DocumentQuestionAnsweringOutputElement,
     DocumentQuestionAnsweringParameters,
 )
-from .feature_extraction import FeatureExtractionInput
+from .feature_extraction import FeatureExtractionInput, FeatureExtractionInputTruncationDirection
 from .fill_mask import FillMaskInput, FillMaskOutputElement, FillMaskParameters
 from .image_classification import (
     ImageClassificationInput,
@@ -64,7 +67,12 @@ from .image_classification import (
     ImageClassificationOutputTransform,
     ImageClassificationParameters,
 )
-from .image_segmentation import ImageSegmentationInput, ImageSegmentationOutputElement, ImageSegmentationParameters
+from .image_segmentation import (
+    ImageSegmentationInput,
+    ImageSegmentationOutputElement,
+    ImageSegmentationParameters,
+    ImageSegmentationSubtask,
+)
 from .image_to_image import ImageToImageInput, ImageToImageOutput, ImageToImageParameters, ImageToImageTargetSize
 from .image_to_text import (
     ImageToTextEarlyStoppingEnum,
@@ -86,13 +94,25 @@ from .question_answering import (
     QuestionAnsweringParameters,
 )
 from .sentence_similarity import SentenceSimilarityInput, SentenceSimilarityInputData
-from .summarization import SummarizationInput, SummarizationOutput, SummarizationParameters
+from .summarization import (
+    SummarizationInput,
+    SummarizationOutput,
+    SummarizationParameters,
+    SummarizationTruncationStrategy,
+)
 from .table_question_answering import (
+    Padding,
     TableQuestionAnsweringInput,
     TableQuestionAnsweringInputData,
     TableQuestionAnsweringOutputElement,
+    TableQuestionAnsweringParameters,
+)
+from .text2text_generation import (
+    Text2TextGenerationInput,
+    Text2TextGenerationOutput,
+    Text2TextGenerationParameters,
+    Text2TextGenerationTruncationStrategy,
 )
-from .text2text_generation import Text2TextGenerationInput, Text2TextGenerationOutput, Text2TextGenerationParameters
 from .text_classification import (
     TextClassificationInput,
     TextClassificationOutputElement,
@@ -106,11 +126,13 @@ from .text_generation import (
     TextGenerationOutput,
     TextGenerationOutputBestOfSequence,
     TextGenerationOutputDetails,
+    TextGenerationOutputFinishReason,
     TextGenerationOutputPrefillToken,
     TextGenerationOutputToken,
     TextGenerationStreamOutput,
     TextGenerationStreamOutputStreamDetails,
     TextGenerationStreamOutputToken,
+    TypeEnum,
 )
 from .text_to_audio import (
     TextToAudioEarlyStoppingEnum,
@@ -128,11 +150,12 @@ from .text_to_speech import (
     TextToSpeechParameters,
 )
 from .token_classification import (
+    TokenClassificationAggregationStrategy,
     TokenClassificationInput,
     TokenClassificationOutputElement,
     TokenClassificationParameters,
 )
-from .translation import TranslationInput, TranslationOutput, TranslationParameters
+from .translation import TranslationInput, TranslationOutput, TranslationParameters, TranslationTruncationStrategy
 from .video_classification import (
     VideoClassificationInput,
     VideoClassificationOutputElement,
@@ -147,19 +170,17 @@ from .visual_question_answering import (
 )
 from .zero_shot_classification import (
     ZeroShotClassificationInput,
-    ZeroShotClassificationInputData,
     ZeroShotClassificationOutputElement,
     ZeroShotClassificationParameters,
 )
 from .zero_shot_image_classification import (
     ZeroShotImageClassificationInput,
-    ZeroShotImageClassificationInputData,
     ZeroShotImageClassificationOutputElement,
     ZeroShotImageClassificationParameters,
 )
 from .zero_shot_object_detection import (
     ZeroShotObjectDetectionBoundingBox,
     ZeroShotObjectDetectionInput,
-    ZeroShotObjectDetectionInputData,
     ZeroShotObjectDetectionOutputElement,
+    ZeroShotObjectDetectionParameters,
 )

huggingface_hub/inference/_generated/types/audio_classification.py

@@ -14,12 +14,10 @@ AudioClassificationOutputTransform = Literal["sigmoid", "softmax", "none"]
 
 @dataclass
 class AudioClassificationParameters(BaseInferenceType):
-    """Additional inference parameters
-    Additional inference parameters for Audio Classification
-    """
+    """Additional inference parameters for Audio Classification"""
 
     function_to_apply: Optional["AudioClassificationOutputTransform"] = None
-    """The function to apply to the output."""
+    """The function to apply to the model outputs in order to retrieve the scores."""
     top_k: Optional[int] = None
     """When specified, limits the output to the top K most probable classes."""
 
@@ -33,7 +31,7 @@ class AudioClassificationInput(BaseInferenceType):
     also provide the audio data as a raw bytes payload.
     """
     parameters: Optional[AudioClassificationParameters] = None
-    """Additional inference parameters"""
+    """Additional inference parameters for Audio Classification"""
 
 
 @dataclass

huggingface_hub/inference/_generated/types/automatic_speech_recognition.py

@@ -14,9 +14,7 @@ AutomaticSpeechRecognitionEarlyStoppingEnum = Literal["never"]
 
 @dataclass
 class AutomaticSpeechRecognitionGenerationParameters(BaseInferenceType):
-    """Parametrization of the text generation process
-    Ad-hoc parametrization of the text generation process
-    """
+    """Parametrization of the text generation process"""
 
     do_sample: Optional[bool] = None
     """Whether to use sampling instead of greedy decoding when generating new tokens."""
@@ -76,11 +74,9 @@ class AutomaticSpeechRecognitionGenerationParameters(BaseInferenceType):
 
 @dataclass
 class AutomaticSpeechRecognitionParameters(BaseInferenceType):
-    """Additional inference parameters
-    Additional inference parameters for Automatic Speech Recognition
-    """
+    """Additional inference parameters for Automatic Speech Recognition"""
 
-    generate: Optional[AutomaticSpeechRecognitionGenerationParameters] = None
+    generation_parameters: Optional[AutomaticSpeechRecognitionGenerationParameters] = None
     """Parametrization of the text generation process"""
     return_timestamps: Optional[bool] = None
     """Whether to output corresponding timestamps with the generated text"""
@@ -95,7 +91,7 @@ class AutomaticSpeechRecognitionInput(BaseInferenceType):
     also provide the audio data as a raw bytes payload.
     """
     parameters: Optional[AutomaticSpeechRecognitionParameters] = None
-    """Additional inference parameters"""
+    """Additional inference parameters for Automatic Speech Recognition"""
 
 
 @dataclass

huggingface_hub/inference/_generated/types/chat_completion.py

@@ -60,8 +60,11 @@ class ChatCompletionInputFunctionName(BaseInferenceType):
 
 
 @dataclass
-class ChatCompletionInputToolType(BaseInferenceType):
-    function: Optional[ChatCompletionInputFunctionName] = None
+class ChatCompletionInputToolChoiceClass(BaseInferenceType):
+    function: ChatCompletionInputFunctionName
+
+
+ChatCompletionInputToolChoiceEnum = Literal["auto", "none", "required"]
 
 
 @dataclass
@@ -72,7 +75,7 @@ class ChatCompletionInputFunctionDefinition(BaseInferenceType):
 
 
 @dataclass
-class ToolElement(BaseInferenceType):
+class ChatCompletionInputTool(BaseInferenceType):
     function: ChatCompletionInputFunctionDefinition
     type: str
 
@@ -138,10 +141,10 @@ class ChatCompletionInput(BaseInferenceType):
     lower values like 0.2 will make it more focused and deterministic.
     We generally recommend altering this or `top_p` but not both.
     """
-    tool_choice: Optional[Union[ChatCompletionInputToolType, str]] = None
+    tool_choice: Optional[Union[ChatCompletionInputToolChoiceClass, "ChatCompletionInputToolChoiceEnum"]] = None
     tool_prompt: Optional[str] = None
     """A prompt to be appended before the tools"""
-    tools: Optional[List[ToolElement]] = None
+    tools: Optional[List[ChatCompletionInputTool]] = None
     """A list of tools the model may call. Currently, only functions are supported as a tool.
     Use this to provide a list of
     functions the model may generate JSON inputs for.

huggingface_hub/inference/_generated/types/depth_estimation.py

@@ -16,7 +16,7 @@ class DepthEstimationInput(BaseInferenceType):
     inputs: Any
     """The input image data"""
     parameters: Optional[Dict[str, Any]] = None
-    """Additional inference parameters"""
+    """Additional inference parameters for Depth Estimation"""
 
 
 @dataclass

huggingface_hub/inference/_generated/types/document_question_answering.py

@@ -21,9 +21,7 @@ class DocumentQuestionAnsweringInputData(BaseInferenceType):
 
 @dataclass
 class DocumentQuestionAnsweringParameters(BaseInferenceType):
-    """Additional inference parameters
-    Additional inference parameters for Document Question Answering
-    """
+    """Additional inference parameters for Document Question Answering"""
 
     doc_stride: Optional[int] = None
     """If the words in the document are too long to fit with the question for the model, it will
@@ -62,7 +60,7 @@ class DocumentQuestionAnsweringInput(BaseInferenceType):
     inputs: DocumentQuestionAnsweringInputData
     """One (document, question) pair to answer"""
     parameters: Optional[DocumentQuestionAnsweringParameters] = None
-    """Additional inference parameters"""
+    """Additional inference parameters for Document Question Answering"""
 
 
 @dataclass
@@ -81,5 +79,3 @@ class DocumentQuestionAnsweringOutputElement(BaseInferenceType):
     """The start word index of the answer (in the OCR’d version of the input or provided word
     boxes).
     """
-    words: List[int]
-    """The index of each word/box pair that is in the answer"""

huggingface_hub/inference/_generated/types/feature_extraction.py

@@ -26,7 +26,7 @@ class FeatureExtractionInput(BaseInferenceType):
     prompt_name: Optional[str] = None
     """The name of the prompt that should be used by for encoding. If not set, no prompt
     will be applied.
-    Must be a key in the `Sentence Transformers` configuration `prompts` dictionary.
+    Must be a key in the `sentence-transformers` configuration `prompts` dictionary.
     For example if ``prompt_name`` is "query" and the ``prompts`` is {"query": "query: ",
     ...},
     then the sentence "What is the capital of France?" will be encoded as

huggingface_hub/inference/_generated/types/fill_mask.py

@@ -11,9 +11,7 @@ from .base import BaseInferenceType
 
 @dataclass
 class FillMaskParameters(BaseInferenceType):
-    """Additional inference parameters
-    Additional inference parameters for Fill Mask
-    """
+    """Additional inference parameters for Fill Mask"""
 
     targets: Optional[List[str]] = None
     """When passed, the model will limit the scores to the passed targets instead of looking up
@@ -32,7 +30,7 @@ class FillMaskInput(BaseInferenceType):
     inputs: str
     """The text with masked tokens"""
     parameters: Optional[FillMaskParameters] = None
-    """Additional inference parameters"""
+    """Additional inference parameters for Fill Mask"""
 
 
 @dataclass

huggingface_hub/inference/_generated/types/image_classification.py

@@ -14,12 +14,10 @@ ImageClassificationOutputTransform = Literal["sigmoid", "softmax", "none"]
 
 @dataclass
 class ImageClassificationParameters(BaseInferenceType):
-    """Additional inference parameters
-    Additional inference parameters for Image Classification
-    """
+    """Additional inference parameters for Image Classification"""
 
     function_to_apply: Optional["ImageClassificationOutputTransform"] = None
-    """The function to apply to the output."""
+    """The function to apply to the model outputs in order to retrieve the scores."""
     top_k: Optional[int] = None
     """When specified, limits the output to the top K most probable classes."""
 
@@ -33,7 +31,7 @@ class ImageClassificationInput(BaseInferenceType):
     also provide the image data as a raw bytes payload.
     """
     parameters: Optional[ImageClassificationParameters] = None
-    """Additional inference parameters"""
+    """Additional inference parameters for Image Classification"""
 
 
 @dataclass

huggingface_hub/inference/_generated/types/image_segmentation.py

@@ -14,9 +14,7 @@ ImageSegmentationSubtask = Literal["instance", "panoptic", "semantic"]
 
 @dataclass
 class ImageSegmentationParameters(BaseInferenceType):
-    """Additional inference parameters
-    Additional inference parameters for Image Segmentation
-    """
+    """Additional inference parameters for Image Segmentation"""
 
     mask_threshold: Optional[float] = None
     """Threshold to use when turning the predicted masks into binary values."""
@@ -37,7 +35,7 @@ class ImageSegmentationInput(BaseInferenceType):
     also provide the image data as a raw bytes payload.
     """
     parameters: Optional[ImageSegmentationParameters] = None
-    """Additional inference parameters"""
+    """Additional inference parameters for Image Segmentation"""
 
 
 @dataclass

huggingface_hub/inference/_generated/types/image_to_image.py

@@ -19,9 +19,7 @@ class ImageToImageTargetSize(BaseInferenceType):
 
 @dataclass
 class ImageToImageParameters(BaseInferenceType):
-    """Additional inference parameters
-    Additional inference parameters for Image To Image
-    """
+    """Additional inference parameters for Image To Image"""
 
     guidance_scale: Optional[float] = None
     """For diffusion models. A higher guidance scale value encourages the model to generate
@@ -46,7 +44,7 @@ class ImageToImageInput(BaseInferenceType):
     also provide the image data as a raw bytes payload.
     """
     parameters: Optional[ImageToImageParameters] = None
-    """Additional inference parameters"""
+    """Additional inference parameters for Image To Image"""
 
 
 @dataclass

huggingface_hub/inference/_generated/types/image_to_text.py

@@ -14,9 +14,7 @@ ImageToTextEarlyStoppingEnum = Literal["never"]
 
 @dataclass
 class ImageToTextGenerationParameters(BaseInferenceType):
-    """Parametrization of the text generation process
-    Ad-hoc parametrization of the text generation process
-    """
+    """Parametrization of the text generation process"""
 
     do_sample: Optional[bool] = None
     """Whether to use sampling instead of greedy decoding when generating new tokens."""
@@ -76,11 +74,9 @@ class ImageToTextGenerationParameters(BaseInferenceType):
 
 @dataclass
 class ImageToTextParameters(BaseInferenceType):
-    """Additional inference parameters
-    Additional inference parameters for Image To Text
-    """
+    """Additional inference parameters for Image To Text"""
 
-    generate: Optional[ImageToTextGenerationParameters] = None
+    generation_parameters: Optional[ImageToTextGenerationParameters] = None
     """Parametrization of the text generation process"""
     max_new_tokens: Optional[int] = None
     """The amount of maximum tokens to generate."""
@@ -93,7 +89,7 @@ class ImageToTextInput(BaseInferenceType):
     inputs: Any
     """The input image data"""
     parameters: Optional[ImageToTextParameters] = None
-    """Additional inference parameters"""
+    """Additional inference parameters for Image To Text"""
 
 
 @dataclass

huggingface_hub/inference/_generated/types/object_detection.py

@@ -11,9 +11,7 @@ from .base import BaseInferenceType
 
 @dataclass
 class ObjectDetectionParameters(BaseInferenceType):
-    """Additional inference parameters
-    Additional inference parameters for Object Detection
-    """
+    """Additional inference parameters for Object Detection"""
 
     threshold: Optional[float] = None
     """The probability necessary to make a prediction."""
@@ -28,7 +26,7 @@ class ObjectDetectionInput(BaseInferenceType):
     also provide the image data as a raw bytes payload.
     """
     parameters: Optional[ObjectDetectionParameters] = None
-    """Additional inference parameters"""
+    """Additional inference parameters for Object Detection"""
 
 
 @dataclass

huggingface_hub/inference/_generated/types/question_answering.py

@@ -21,9 +21,7 @@ class QuestionAnsweringInputData(BaseInferenceType):
 
 @dataclass
 class QuestionAnsweringParameters(BaseInferenceType):
-    """Additional inference parameters
-    Additional inference parameters for Question Answering
-    """
+    """Additional inference parameters for Question Answering"""
 
     align_to_words: Optional[bool] = None
     """Attempts to align the answer to real words. Improves quality on space separated
@@ -60,7 +58,7 @@ class QuestionAnsweringInput(BaseInferenceType):
     inputs: QuestionAnsweringInputData
     """One (context, question) pair to answer"""
     parameters: Optional[QuestionAnsweringParameters] = None
-    """Additional inference parameters"""
+    """Additional inference parameters for Question Answering"""
 
 
 @dataclass

huggingface_hub/inference/_generated/types/sentence_similarity.py

@@ -25,4 +25,4 @@ class SentenceSimilarityInput(BaseInferenceType):
 
     inputs: SentenceSimilarityInputData
     parameters: Optional[Dict[str, Any]] = None
-    """Additional inference parameters"""
+    """Additional inference parameters for Sentence Similarity"""

huggingface_hub/inference/_generated/types/summarization.py

@@ -14,9 +14,7 @@ SummarizationTruncationStrategy = Literal["do_not_truncate", "longest_first", "o
 
 @dataclass
 class SummarizationParameters(BaseInferenceType):
-    """Additional inference parameters.
-    Additional inference parameters for summarization.
-    """
+    """Additional inference parameters for summarization."""
 
     clean_up_tokenization_spaces: Optional[bool] = None
     """Whether to clean up the potential extra spaces in the text output."""
@@ -33,7 +31,7 @@ class SummarizationInput(BaseInferenceType):
     inputs: str
     """The input text to summarize."""
     parameters: Optional[SummarizationParameters] = None
-    """Additional inference parameters."""
+    """Additional inference parameters for summarization."""
 
 
 @dataclass

huggingface_hub/inference/_generated/types/table_question_answering.py

@@ -4,7 +4,7 @@
 #   - script: https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-codegen.ts
 #   - specs:  https://github.com/huggingface/huggingface.js/tree/main/packages/tasks/src/tasks.
 from dataclasses import dataclass
-from typing import Any, Dict, List, Optional
+from typing import Dict, List, Literal, Optional
 
 from .base import BaseInferenceType
 
@@ -19,14 +19,32 @@ class TableQuestionAnsweringInputData(BaseInferenceType):
     """The table to serve as context for the questions"""
 
 
+Padding = Literal["do_not_pad", "longest", "max_length"]
+
+
+@dataclass
+class TableQuestionAnsweringParameters(BaseInferenceType):
+    """Additional inference parameters for Table Question Answering"""
+
+    padding: Optional["Padding"] = None
+    """Activates and controls padding."""
+    sequential: Optional[bool] = None
+    """Whether to do inference sequentially or as a batch. Batching is faster, but models like
+    SQA require the inference to be done sequentially to extract relations within sequences,
+    given their conversational nature.
+    """
+    truncation: Optional[bool] = None
+    """Activates and controls truncation."""
+
+
 @dataclass
 class TableQuestionAnsweringInput(BaseInferenceType):
     """Inputs for Table Question Answering inference"""
 
     inputs: TableQuestionAnsweringInputData
     """One (table, question) pair to answer"""
-    parameters: Optional[Dict[str, Any]] = None
-    """Additional inference parameters"""
+    parameters: Optional[TableQuestionAnsweringParameters] = None
+    """Additional inference parameters for Table Question Answering"""
 
 
 @dataclass

huggingface_hub/inference/_generated/types/text2text_generation.py

@@ -14,9 +14,7 @@ Text2TextGenerationTruncationStrategy = Literal["do_not_truncate", "longest_firs
 
 @dataclass
 class Text2TextGenerationParameters(BaseInferenceType):
-    """Additional inference parameters
-    Additional inference parameters for Text2text Generation
-    """
+    """Additional inference parameters for Text2text Generation"""
 
     clean_up_tokenization_spaces: Optional[bool] = None
     """Whether to clean up the potential extra spaces in the text output."""
@@ -33,7 +31,7 @@ class Text2TextGenerationInput(BaseInferenceType):
     inputs: str
     """The input text data"""
     parameters: Optional[Text2TextGenerationParameters] = None
-    """Additional inference parameters"""
+    """Additional inference parameters for Text2text Generation"""
 
 
 @dataclass

huggingface_hub/inference/_generated/types/text_classification.py

@@ -14,18 +14,12 @@ TextClassificationOutputTransform = Literal["sigmoid", "softmax", "none"]
 
 @dataclass
 class TextClassificationParameters(BaseInferenceType):
-    """
-    Additional inference parameters for Text Classification.
-    """
+    """Additional inference parameters for Text Classification"""
 
     function_to_apply: Optional["TextClassificationOutputTransform"] = None
-    """
-    The function to apply to the output.
-    """
+    """The function to apply to the model outputs in order to retrieve the scores."""
     top_k: Optional[int] = None
-    """
-    When specified, limits the output to the top K most probable classes.
-    """
+    """When specified, limits the output to the top K most probable classes."""
 
 
 @dataclass
@@ -35,7 +29,7 @@ class TextClassificationInput(BaseInferenceType):
     inputs: str
     """The text to classify"""
     parameters: Optional[TextClassificationParameters] = None
-    """Additional inference parameters"""
+    """Additional inference parameters for Text Classification"""
 
 
 @dataclass

huggingface_hub/inference/_generated/types/text_to_audio.py

@@ -14,9 +14,7 @@ TextToAudioEarlyStoppingEnum = Literal["never"]
 
 @dataclass
 class TextToAudioGenerationParameters(BaseInferenceType):
-    """Parametrization of the text generation process
-    Ad-hoc parametrization of the text generation process
-    """
+    """Parametrization of the text generation process"""
 
     do_sample: Optional[bool] = None
     """Whether to use sampling instead of greedy decoding when generating new tokens."""
@@ -40,11 +38,11 @@ class TextToAudioGenerationParameters(BaseInferenceType):
     max_length: Optional[int] = None
     """The maximum length (in tokens) of the generated text, including the input."""
     max_new_tokens: Optional[int] = None
-    """The maximum number of tokens to generate. Takes precedence over maxLength."""
+    """The maximum number of tokens to generate. Takes precedence over max_length."""
     min_length: Optional[int] = None
     """The minimum length (in tokens) of the generated text, including the input."""
     min_new_tokens: Optional[int] = None
-    """The minimum number of tokens to generate. Takes precedence over maxLength."""
+    """The minimum number of tokens to generate. Takes precedence over min_length."""
     num_beam_groups: Optional[int] = None
     """Number of groups to divide num_beams into in order to ensure diversity among different
     groups of beams. See [this paper](https://hf.co/papers/1610.02424) for more details.
@@ -76,11 +74,9 @@ class TextToAudioGenerationParameters(BaseInferenceType):
 
 @dataclass
 class TextToAudioParameters(BaseInferenceType):
-    """Additional inference parameters
-    Additional inference parameters for Text To Audio
-    """
+    """Additional inference parameters for Text To Audio"""
 
-    generate: Optional[TextToAudioGenerationParameters] = None
+    generation_parameters: Optional[TextToAudioGenerationParameters] = None
     """Parametrization of the text generation process"""
 
 
@@ -91,7 +87,7 @@ class TextToAudioInput(BaseInferenceType):
     inputs: str
     """The input text data"""
     parameters: Optional[TextToAudioParameters] = None
-    """Additional inference parameters"""
+    """Additional inference parameters for Text To Audio"""
 
 
 @dataclass

huggingface_hub/inference/_generated/types/text_to_image.py

@@ -19,9 +19,7 @@ class TextToImageTargetSize(BaseInferenceType):
 
 @dataclass
 class TextToImageParameters(BaseInferenceType):
-    """Additional inference parameters
-    Additional inference parameters for Text To Image
-    """
+    """Additional inference parameters for Text To Image"""
 
     guidance_scale: Optional[float] = None
     """A higher guidance scale value encourages the model to generate images closely linked to
@@ -48,7 +46,7 @@ class TextToImageInput(BaseInferenceType):
     inputs: str
     """The input text data (sometimes called "prompt")"""
     parameters: Optional[TextToImageParameters] = None
-    """Additional inference parameters"""
+    """Additional inference parameters for Text To Image"""
 
 
 @dataclass