occurs automatically when you want to use the result of a computation that has not yet been materialized.   Copied %%time
images = p_generate(prompt_ids, p_params, rng)
images = images.block_until_ready()

# CPU times: user 1min 15s, sys: 18.2 s, total: 1min 34s
# Wall time: 1min 15s Check your image dimensions to see if they’re correct:   Copied images.shape
# (8, 1, 512, 512, 3)  Resources To learn more about how JAX works with Stable Diffusi