input_ids = torch.tensor([list("Life is like a box of chocolates.