_pretrained("runwayml/stable-diffusion-v1-5")
>>> # Model was saved using *save_pretrained('./test/saved_model/')* (for example purposes, not runnable).
>>> model, params = FlaxUNet2DConditionModel.from_pretrained("./test/saved_model/")  If you get the error message below, you need to finetune the weights for your downstream task:   Copied Some weights of UNet2DConditionModel were not initialized from the model checkpoint at runwayml/stable-dif