SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

correct TTS pipeline docstrings snippet (#25587)

Browse Source 
* correct TTS pipeline docstrings snippet

* add text_to_audio.py pipelines to documentation tests
This commit is contained in:
Yoach Lacombe
2023-08-21 13:40:04 +02:00
committed by GitHub
parent e769ca3d28
commit 2c1bcbf5ed
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/pipelines/text_to_audio.py
View File

@@ -34,8 +34,8 @@ class TextToAudioPipeline(Pipeline):
```python ```python
>>> from transformers import pipeline >>> from transformers import pipeline
>>> classifier = pipeline(model="suno/bark") >>> pipe = pipeline(model="suno/bark-small")
>>> output = pipeline("Hey it's HuggingFace on the phone!") >>> output = pipe("Hey it's HuggingFace on the phone!")
>>> audio = output["audio"] >>> audio = output["audio"]
>>> sampling_rate = output["sampling_rate"] >>> sampling_rate = output["sampling_rate"]

1
utils/documentation_tests.txt
View File

@@ -489,3 +489,4 @@ src/transformers/models/yolos/image_processing_yolos.py
src/transformers/models/yolos/modeling_yolos.py src/transformers/models/yolos/modeling_yolos.py
src/transformers/models/yoso/configuration_yoso.py src/transformers/models/yoso/configuration_yoso.py
src/transformers/pipelines/ src/transformers/pipelines/
src/transformers/pipelines/text_to_audio.py