def METHOD_NAME(self):
    prompt = "a caterpillar smoking a hookah while sitting on a mushroom"
    stablediff = StableDiffusion(128, 128)
    img = stablediff.text_to_image(prompt, seed=1337, num_steps=5)
    # Verify that the step-by-step creation flow creates an identical output
    text_encoding = stablediff.encode_text(prompt)
    self.assertAllClose(
        img,
        stablediff.generate_image(text_encoding, seed=1337, num_steps=5),
        atol=1e-4,
    )