[Flux LoRA] fix issues in flux lora scripts #11111
linoytsaban
merged
33 commits into
huggingface:main
from
linoytsaban:flux_lora_advanced
remove custom scheduler
298709e1
update requirements.txt
364f478d
log_validation with mixed precision
90e9517f
linoytsaban
changed the title
Merge branch 'main' into flux_lora_advanced
d90d7f02
add intermediate embeddings saving when checkpointing is enabled
bdd6caef
remove comment
c8e165b5
Merge remote-tracking branch 'origin/flux_lora_advanced' into flux_lo…
d434db37
fix validation
710fcae4
Merge branch 'main' into flux_lora_advanced
2d8ca606
add unwrap_model for accelerator, torch.no_grad context for validatio…
0565932e
revert unwrap_model change temp
ba4decee
add .module to address distributed training bug + replace accelerator…
c155f22a
changes to align advanced script with canonical script
9c4368d8
make changes for distributed training + unify unwrap_model calls in a…
7492e925
add module.dtype fix to dreambooth script
0729c662
unify unwrap_model calls in dreambooth script
cc1d2ad8
Merge branch 'main' into flux_lora_advanced
07c29747
Merge branch 'main' into flux_lora_advanced
603b57ce
Merge branch 'main' into flux_lora_advanced
e5636f0e
fix condition in validation run
8bf49c77
Merge branch 'main' into flux_lora_advanced
b211eead
mixed precision
9b2917fd
Merge branch 'main' into flux_lora_advanced
22046d15
Merge branch 'main' into flux_lora_advanced
f1af7e22
Merge branch 'main' into flux_lora_advanced
8dc70050
Merge branch 'main' into flux_lora_advanced
d8ef75fc
Merge branch 'main' into flux_lora_advanced
bfd8d45e
Update examples/advanced_diffusion_training/train_dreambooth_lora_flu…
5d249a77
Merge branch 'main' into flux_lora_advanced
a4b1e7fe
smol style change
57ee3cf0
change autocast
8b991a5c
sayakpaul
approved these changes
on 2025-04-08
![github-actions[bot]](https://avatars.githubusercontent.com/in/15368?v=4&s=60)
Apply style fixes
bfd1df62
Merge branch 'main' into flux_lora_advanced
c978ca37
Login to write a write a comment.
Login via GitHub