Function prepare_bart_inputs_dict

tests/test_modeling_bart.py:101–109

(
    config, input_ids, attention_mask=None,
)

def prepare_bart_inputs_dict(
    config, input_ids, attention_mask=None,
):
    if attention_mask is None:
        attention_mask = input_ids.ne(config.pad_token_id)
    return {
        "input_ids": input_ids,
        "attention_mask": attention_mask,
    }


@require_torch