We hadn't yet added in a parse_inputs function to the new TextGenerationPipeline.
The base parse_inputs function previously in BasePipeline is covered by the normal operator functionality but for TextGeneration, we do a bit of extra kwarg/arg parsing. This operator handles that additional functionality
Summary
parse_inputs
function to the newTextGenerationPipeline
.parse_inputs
function previously inBasePipeline
is covered by the normal operator functionality but for TextGeneration, we do a bit of extra kwarg/arg parsing. This operator handles that additional functionalityTesting
Examples of cases which now work:
With the following pipeline:
Not explicitly listing
prompt
orsequences
Listing
sequences
:Listing
prompt
:Passing generation parameters as kwargs:
Passing
generation_config
and overriding with kwargs: