pipeline, params = FlaxStableDiffusionImg2ImgPipeline.from_pretrained(
"CompVis/stable-diffusion-v1-4",
revision="flax",
dtype=jnp.bfloat16,
safety_checker=None,
load_attn_procs="sayakpaul/civitai-light-shadow-lora",
)
The above code does nothing.
I cannot use load_attn_procs
or load_lora_weights
with unet
or pipeline
as they do not exist on FlaxStableDiffusionImg2ImgPipeline.
How can I use LoRA with it then?