fied in the command. To load the fine-tuned model for inference, just pass that path to StableDiffusionPipeline:


	Copied
from diffusers import StableDiffusionPipeline

model_path = "path_to_saved_model"
pipe = StableDiffusionPipeline.from_pretrained(model_path, torch_dtype=torch.float16)
pipe.to("cuda")

image = pipe(prompt="yoda").images[0]
image.save("yoda-pokemon.png")

Flax / JAX fine-tuning
	
Thanks to @duongna211 it’s possible to fine-t