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

[pipelines] Text2TextGenerationPipeline (#6744)

Browse Source 
* add Text2TextGenerationPipeline

* remove max length warning

* remove comments

* remove input_length

* fix typo

* add tests

* use TFAutoModelForSeq2SeqLM

* doc

* typo

* add the doc below TextGenerationPipeline

* doc nit

* style

* delete comment
This commit is contained in:
Suraj Patil
2020-09-02 17:04:35 +05:30
committed by GitHub
parent 6b24281229
commit 4230d30f77
4 changed files with 139 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
docs/source/main_classes/pipelines.rst
View File

@@ -21,6 +21,7 @@ There are two categories of pipeline abstractions to be aware about:
- :class:`~transformers.TokenClassificationPipeline`
- :class:`~transformers.TranslationPipeline`
- :class:`~transformers.ZeroShotClassificationPipeline`
- :class:`~transformers.Text2TextGenerationPipeline`
The pipeline abstraction
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
@@ -91,6 +92,13 @@ TextGenerationPipeline
:special-members: __call__
:members:
Text2TextGenerationPipeline
==========================================
.. autoclass:: transformers.Text2TextGenerationPipeline
:special-members: __call__
:members:
TokenClassificationPipeline
==========================================
@@ -105,7 +113,6 @@ ZeroShotClassificationPipeline
:special-members: __call__
:members:
Parent class: :obj:`Pipeline`
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~