Fix shared classes imports + text-to-speech task

huggingface · Sep 27, 2024 · 4fd4aba · 4fd4aba
1 parent aa8abc0
commit 4fd4aba
Show file tree

Hide file tree

Showing 16 changed files with 337 additions and 61 deletions.
diff --git a/docs/source/en/package_reference/inference_types.md b/docs/source/en/package_reference/inference_types.md
@@ -23,6 +23,8 @@ This part of the lib is still under development and will be improved in future r
 
 [[autodoc]] huggingface_hub.AudioClassificationOutputElement
 
+[[autodoc]] huggingface_hub.AudioClassificationOutputTransform
+
 [[autodoc]] huggingface_hub.AudioClassificationParameters
 
 
@@ -37,6 +39,8 @@ This part of the lib is still under development and will be improved in future r
 
 ## automatic_speech_recognition
 
+[[autodoc]] huggingface_hub.AutomaticSpeechRecognitionEarlyStoppingEnum
+
 [[autodoc]] huggingface_hub.AutomaticSpeechRecognitionGenerationParameters
 
 [[autodoc]] huggingface_hub.AutomaticSpeechRecognitionInput
@@ -151,6 +155,8 @@ This part of the lib is still under development and will be improved in future r
 
 [[autodoc]] huggingface_hub.ImageClassificationOutputElement
 
+[[autodoc]] huggingface_hub.ImageClassificationOutputTransform
+
 [[autodoc]] huggingface_hub.ImageClassificationParameters
 
 
@@ -179,6 +185,8 @@ This part of the lib is still under development and will be improved in future r
 
 ## image_to_text
 
+[[autodoc]] huggingface_hub.ImageToTextEarlyStoppingEnum
+
 [[autodoc]] huggingface_hub.ImageToTextGenerationParameters
 
 [[autodoc]] huggingface_hub.ImageToTextInput
@@ -257,6 +265,8 @@ This part of the lib is still under development and will be improved in future r
 
 [[autodoc]] huggingface_hub.TextClassificationOutputElement
 
+[[autodoc]] huggingface_hub.TextClassificationOutputTransform
+
 [[autodoc]] huggingface_hub.TextClassificationParameters
 
 
@@ -289,6 +299,8 @@ This part of the lib is still under development and will be improved in future r
 
 ## text_to_audio
 
+[[autodoc]] huggingface_hub.TextToAudioEarlyStoppingEnum
+
 [[autodoc]] huggingface_hub.TextToAudioGenerationParameters
 
 [[autodoc]] huggingface_hub.TextToAudioInput
@@ -313,6 +325,8 @@ This part of the lib is still under development and will be improved in future r
 
 ## text_to_speech
 
+[[autodoc]] huggingface_hub.TextToSpeechEarlyStoppingEnum
+
 [[autodoc]] huggingface_hub.TextToSpeechGenerationParameters
 
 [[autodoc]] huggingface_hub.TextToSpeechInput
@@ -349,6 +363,8 @@ This part of the lib is still under development and will be improved in future r
 
 [[autodoc]] huggingface_hub.VideoClassificationOutputElement
 
+[[autodoc]] huggingface_hub.VideoClassificationOutputTransform
+
 [[autodoc]] huggingface_hub.VideoClassificationParameters
 
 
@@ -398,5 +414,3 @@ This part of the lib is still under development and will be improved in future r
 [[autodoc]] huggingface_hub.ZeroShotObjectDetectionInputData
 
 [[autodoc]] huggingface_hub.ZeroShotObjectDetectionOutputElement
-
-
diff --git a/docs/source/ko/package_reference/inference_types.md b/docs/source/ko/package_reference/inference_types.md
@@ -22,6 +22,8 @@ rendered properly in your Markdown viewer.
 
 [[autodoc]] huggingface_hub.AudioClassificationOutputElement
 
+[[autodoc]] huggingface_hub.AudioClassificationOutputTransform
+
 [[autodoc]] huggingface_hub.AudioClassificationParameters
 
 
@@ -34,7 +36,9 @@ rendered properly in your Markdown viewer.
 
 
 
-## automatic_speech_recognition[[huggingface_hub.AutomaticSpeechRecognitionGenerationParameters]]
+## automatic_speech_recognition[[huggingface_hub.AutomaticSpeechRecognitionEarlyStoppingEnum]]
+
+[[autodoc]] huggingface_hub.AutomaticSpeechRecognitionEarlyStoppingEnum
 
 [[autodoc]] huggingface_hub.AutomaticSpeechRecognitionGenerationParameters
 
@@ -150,6 +154,8 @@ rendered properly in your Markdown viewer.
 
 [[autodoc]] huggingface_hub.ImageClassificationOutputElement
 
+[[autodoc]] huggingface_hub.ImageClassificationOutputTransform
+
 [[autodoc]] huggingface_hub.ImageClassificationParameters
 
 
@@ -176,7 +182,9 @@ rendered properly in your Markdown viewer.
 
 
 
-## image_to_text[[huggingface_hub.ImageToTextGenerationParameters]]
+## image_to_text[[huggingface_hub.ImageToTextEarlyStoppingEnum]]
+
+[[autodoc]] huggingface_hub.ImageToTextEarlyStoppingEnum
 
 [[autodoc]] huggingface_hub.ImageToTextGenerationParameters
 
@@ -256,6 +264,8 @@ rendered properly in your Markdown viewer.
 
 [[autodoc]] huggingface_hub.TextClassificationOutputElement
 
+[[autodoc]] huggingface_hub.TextClassificationOutputTransform
+
 [[autodoc]] huggingface_hub.TextClassificationParameters
 
 
@@ -286,7 +296,9 @@ rendered properly in your Markdown viewer.
 
 
 
-## text_to_audio[[huggingface_hub.TextToAudioGenerationParameters]]
+## text_to_audio[[huggingface_hub.TextToAudioEarlyStoppingEnum]]
+
+[[autodoc]] huggingface_hub.TextToAudioEarlyStoppingEnum
 
 [[autodoc]] huggingface_hub.TextToAudioGenerationParameters
 
@@ -310,7 +322,9 @@ rendered properly in your Markdown viewer.
 
 
 
-## text_to_speech[[huggingface_hub.TextToSpeechGenerationParameters]]
+## text_to_speech[[huggingface_hub.TextToSpeechEarlyStoppingEnum]]
+
+[[autodoc]] huggingface_hub.TextToSpeechEarlyStoppingEnum
 
 [[autodoc]] huggingface_hub.TextToSpeechGenerationParameters
 
@@ -348,6 +362,8 @@ rendered properly in your Markdown viewer.
 
 [[autodoc]] huggingface_hub.VideoClassificationOutputElement
 
+[[autodoc]] huggingface_hub.VideoClassificationOutputTransform
+
 [[autodoc]] huggingface_hub.VideoClassificationParameters
 
 
@@ -397,5 +413,3 @@ rendered properly in your Markdown viewer.
 [[autodoc]] huggingface_hub.ZeroShotObjectDetectionInputData
 
 [[autodoc]] huggingface_hub.ZeroShotObjectDetectionOutputElement
-
-
diff --git a/src/huggingface_hub/__init__.py b/src/huggingface_hub/__init__.py
@@ -280,9 +280,11 @@
     "inference._generated.types": [
         "AudioClassificationInput",
         "AudioClassificationOutputElement",
+        "AudioClassificationOutputTransform",
         "AudioClassificationParameters",
         "AudioToAudioInput",
         "AudioToAudioOutputElement",
+        "AutomaticSpeechRecognitionEarlyStoppingEnum",
         "AutomaticSpeechRecognitionGenerationParameters",
         "AutomaticSpeechRecognitionInput",
         "AutomaticSpeechRecognitionOutput",
@@ -327,6 +329,7 @@
         "FillMaskParameters",
         "ImageClassificationInput",
         "ImageClassificationOutputElement",
+        "ImageClassificationOutputTransform",
         "ImageClassificationParameters",
         "ImageSegmentationInput",
         "ImageSegmentationOutputElement",
@@ -335,6 +338,7 @@
         "ImageToImageOutput",
         "ImageToImageParameters",
         "ImageToImageTargetSize",
+        "ImageToTextEarlyStoppingEnum",
         "ImageToTextGenerationParameters",
         "ImageToTextInput",
         "ImageToTextOutput",
@@ -360,6 +364,7 @@
         "Text2TextGenerationParameters",
         "TextClassificationInput",
         "TextClassificationOutputElement",
+        "TextClassificationOutputTransform",
         "TextClassificationParameters",
         "TextGenerationInput",
         "TextGenerationInputGenerateParameters",
@@ -372,6 +377,7 @@
         "TextGenerationStreamOutput",
         "TextGenerationStreamOutputStreamDetails",
         "TextGenerationStreamOutputToken",
+        "TextToAudioEarlyStoppingEnum",
         "TextToAudioGenerationParameters",
         "TextToAudioInput",
         "TextToAudioOutput",
@@ -380,6 +386,7 @@
         "TextToImageOutput",
         "TextToImageParameters",
         "TextToImageTargetSize",
+        "TextToSpeechEarlyStoppingEnum",
         "TextToSpeechGenerationParameters",
         "TextToSpeechInput",
         "TextToSpeechOutput",
@@ -393,6 +400,7 @@
         "TranslationParameters",
         "VideoClassificationInput",
         "VideoClassificationOutputElement",
+        "VideoClassificationOutputTransform",
         "VideoClassificationParameters",
         "VisualQuestionAnsweringInput",
         "VisualQuestionAnsweringInputData",
@@ -794,9 +802,11 @@ def __dir__():
     from .inference._generated.types import (
         AudioClassificationInput,  # noqa: F401
         AudioClassificationOutputElement,  # noqa: F401
+        AudioClassificationOutputTransform,  # noqa: F401
         AudioClassificationParameters,  # noqa: F401
         AudioToAudioInput,  # noqa: F401
         AudioToAudioOutputElement,  # noqa: F401
+        AutomaticSpeechRecognitionEarlyStoppingEnum,  # noqa: F401
         AutomaticSpeechRecognitionGenerationParameters,  # noqa: F401
         AutomaticSpeechRecognitionInput,  # noqa: F401
         AutomaticSpeechRecognitionOutput,  # noqa: F401
@@ -841,6 +851,7 @@ def __dir__():
         FillMaskParameters,  # noqa: F401
         ImageClassificationInput,  # noqa: F401
         ImageClassificationOutputElement,  # noqa: F401
+        ImageClassificationOutputTransform,  # noqa: F401
         ImageClassificationParameters,  # noqa: F401
         ImageSegmentationInput,  # noqa: F401
         ImageSegmentationOutputElement,  # noqa: F401
@@ -849,6 +860,7 @@ def __dir__():
         ImageToImageOutput,  # noqa: F401
         ImageToImageParameters,  # noqa: F401
         ImageToImageTargetSize,  # noqa: F401
+        ImageToTextEarlyStoppingEnum,  # noqa: F401
         ImageToTextGenerationParameters,  # noqa: F401
         ImageToTextInput,  # noqa: F401
         ImageToTextOutput,  # noqa: F401
@@ -874,6 +886,7 @@ def __dir__():
         Text2TextGenerationParameters,  # noqa: F401
         TextClassificationInput,  # noqa: F401
         TextClassificationOutputElement,  # noqa: F401
+        TextClassificationOutputTransform,  # noqa: F401
         TextClassificationParameters,  # noqa: F401
         TextGenerationInput,  # noqa: F401
         TextGenerationInputGenerateParameters,  # noqa: F401
@@ -886,6 +899,7 @@ def __dir__():
         TextGenerationStreamOutput,  # noqa: F401
         TextGenerationStreamOutputStreamDetails,  # noqa: F401
         TextGenerationStreamOutputToken,  # noqa: F401
+        TextToAudioEarlyStoppingEnum,  # noqa: F401
         TextToAudioGenerationParameters,  # noqa: F401
         TextToAudioInput,  # noqa: F401
         TextToAudioOutput,  # noqa: F401
@@ -894,6 +908,7 @@ def __dir__():
         TextToImageOutput,  # noqa: F401
         TextToImageParameters,  # noqa: F401
         TextToImageTargetSize,  # noqa: F401
+        TextToSpeechEarlyStoppingEnum,  # noqa: F401
         TextToSpeechGenerationParameters,  # noqa: F401
         TextToSpeechInput,  # noqa: F401
         TextToSpeechOutput,  # noqa: F401
@@ -907,6 +922,7 @@ def __dir__():
         TranslationParameters,  # noqa: F401
         VideoClassificationInput,  # noqa: F401
         VideoClassificationOutputElement,  # noqa: F401
+        VideoClassificationOutputTransform,  # noqa: F401
         VideoClassificationParameters,  # noqa: F401
         VisualQuestionAnsweringInput,  # noqa: F401
         VisualQuestionAnsweringInputData,  # noqa: F401