t=True).images
images = pipeline.numpy_to_pil(np.asarray(images.reshape((num_samples,) + images.shape[-3:])))
The following Flax schedulers are not yet compatible with the Flax Stable Diffusion Pipeline:
FlaxLMSDiscreteScheduler
FlaxDDPMScheduler
