FillBlank#

class recwizard.pipelines.fill_blank.configuration_fill_blank.FillBlankConfig(rec_pattern: str = '<movie>', resp_prompt='System:', **kwargs)[source]#
__init__(rec_pattern: str = '<movie>', resp_prompt='System:', **kwargs)[source]#
Parameters:

  • WEIGHT_DIMENSIONS (dict, optional) – The dimension and dtype of module parameters. Used to initialize the parameters when they are not explicitly specified in module initialization. Defaults to None. See also recwizard.module_utils.BaseModule.prepare_weight().

  • **kwargs – Additional parameters. Will be passed to the PretrainedConfig.__init__.

class recwizard.pipelines.fill_blank.modeling_fill_blank.FillBlankPipeline(config, use_resp=True, **kwargs)[source]#
response(**kwargs)#

The main function for the model to generate a response given a query.

Parameters:

  • query (str) – The formatted dialogue history. See the format at First Example.

  • return_dict (bool) – if set to True, will return a dict of outputs instead of a single output.

  • rec_args (dict) – The arguments passed to the recommender module.

  • gen_args (dict) – The arguments passed to the generator module.