def METHOD_NAME(self):
    prompt = "a caterpillar smoking a hookah while sitting on a mushroom"
    stablediff = StableDiffusion(128, 128)
    text_encoding = stablediff.tokenizer.encode(prompt)
    self.assertEqual(
        text_encoding[0:5],
        [49406, 320, 27111, 9038, 320],
    )