diffusers icon indicating copy to clipboard operation
diffusers copied to clipboard

Published 20 hours ago • verified publisher icon huggingface

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

Open linoytsaban opened this issue 4 months 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