Nicolas Patry c2d0ffec8c Adding a new return_full_text parameter to TextGenerationPipeline. (#9852) ...

* Adding a new `return_full_text` parameter to TextGenerationPipeline.

For text-generation, it's sometimes used as prompting text.
In that context, prefixing `generated_text` with the actual input
forces the caller to take an extra step to remove it.

The proposed change adds a new parameter (for backward compatibility).
`return_full_text` that enables the caller to prevent adding the prefix.

* Doc quality.

2021-01-29 10:27:32 +01:00

2.7 KiB

Raw Blame History

View Raw