From 2c1bcbf5ed2536566bcb52ffe37ff70278309205 Mon Sep 17 00:00:00 2001 From: Yoach Lacombe <52246514+ylacombe@users.noreply.github.com> Date: Mon, 21 Aug 2023 13:40:04 +0200 Subject: [PATCH] correct TTS pipeline docstrings snippet (#25587) * correct TTS pipeline docstrings snippet * add text_to_audio.py pipelines to documentation tests --- src/transformers/pipelines/text_to_audio.py | 4 ++-- utils/documentation_tests.txt | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/transformers/pipelines/text_to_audio.py b/src/transformers/pipelines/text_to_audio.py index 5676aa6c21..f2c7e1dd4d 100644 --- a/src/transformers/pipelines/text_to_audio.py +++ b/src/transformers/pipelines/text_to_audio.py @@ -34,8 +34,8 @@ class TextToAudioPipeline(Pipeline): ```python >>> from transformers import pipeline - >>> classifier = pipeline(model="suno/bark") - >>> output = pipeline("Hey it's HuggingFace on the phone!") + >>> pipe = pipeline(model="suno/bark-small") + >>> output = pipe("Hey it's HuggingFace on the phone!") >>> audio = output["audio"] >>> sampling_rate = output["sampling_rate"] diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt index 1fbbc4ee9b..0649f6e028 100644 --- a/utils/documentation_tests.txt +++ b/utils/documentation_tests.txt @@ -489,3 +489,4 @@ src/transformers/models/yolos/image_processing_yolos.py src/transformers/models/yolos/modeling_yolos.py src/transformers/models/yoso/configuration_yoso.py src/transformers/pipelines/ +src/transformers/pipelines/text_to_audio.py \ No newline at end of file