def METHOD_NAME(
    self,
    config,
    input_ids,
    input_mask,
):
    model = BlenderbotForCausalLM(config)
    model.eval()
    result = model(
        input_ids,
        attention_mask=input_mask,
    )
    self.parent.assertEqual(result.shape, [self.batch_size, self.seq_length, self.vocab_size])