def METHOD_NAME(self):
    """
    Ensures gpt2 provides the same generation results regardless of batchsize.
    """
    for add_start_token in [True, False]:
        for batchsize in [1, 2, 4]:
            with self.subTest(
                f'test_batchsize with bs={batchsize} and ast={add_start_token}'
            ):
                self._test_batchsize(batchsize, add_start_token)