def METHOD_NAME(self):
    stablediff = StableDiffusion(128, 128)
    outputs = stablediff.image_encoder.predict(ops.ones((1, 128, 128, 3)))
    self.assertAllClose(
        outputs[0][1:4][0][0],
        [2.451568, 1.607522, -0.546311, -1.194388],
        atol=5e-4,
    )