def METHOD_NAME(self, config, input_ids, input_mask):
    model = DalleBartForConditionalGeneration(config)
    model.eval()
    result = model(input_ids, attention_mask=input_mask)
    self.parent.assertEqual(result[0].shape, [self.seq_length, self.image_vocab_size + 1])