n parallel.
The same way as when we used the pipeline call, the first time we run the following cell it will take a while, but then it will be much faster.


	Copied
%%time
images = p_generate(prompt_ids, p_params, rng)
images = images.block_until_ready()
images.shape


	Copied
CPU times: user 1min 15s, sys: 18.2 s, total: 1min 34s
Wall time: 1min 15s


	Copied
images.shape


	Copied
(8, 1, 512, 512, 3)
We use block_until_ready() to correctly m