diffusers icon indicating copy to clipboard operation
diffusers copied to clipboard

[flux dreambooth lora training] make LoRA target modules configurable + small bug fix

Open linoytsaban opened this issue 1 year ago • 1 comments

new feature for the Flux dreambooth lora training script:

  • make LoRA target modules configurable through --lora_blocks

  • change the current default target modules to not be attention layers only (?)

& small fix to mixed precision training for dreambooth script, as proposed in https://github.com/huggingface/diffusers/pull/9565

linoytsaban avatar Oct 11 '24 08:10 linoytsaban

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.