mlx-examples/flux/flux
2025-03-24 22:16:48 -07:00
..
__init__.py FLUX: save train config (#1049) 2024-11-08 17:15:19 -08:00
autoencoder.py Add FLUX finetuning (#1028) 2024-10-11 21:17:41 -07:00
clip.py Add FLUX finetuning (#1028) 2024-10-11 21:17:41 -07:00
datasets.py FLUX: Optimize dataset loading logic (#1038) 2024-10-15 10:37:45 -07:00
flux.py FLUX: Optimize dataset loading logic (#1038) 2024-10-15 10:37:45 -07:00
layers.py Distributed FLUX (#1325) 2025-03-24 22:16:48 -07:00
lora.py Add FLUX finetuning (#1028) 2024-10-11 21:17:41 -07:00
model.py Distributed FLUX (#1325) 2025-03-24 22:16:48 -07:00
sampler.py Change Flux default max_shift to 1.15 to match the official one (#1137) 2024-12-08 23:29:48 -08:00
t5.py Add FLUX finetuning (#1028) 2024-10-11 21:17:41 -07:00
tokenizers.py Add FLUX finetuning (#1028) 2024-10-11 21:17:41 -07:00
trainer.py FLUX: Optimize dataset loading logic (#1038) 2024-10-15 10:37:45 -07:00
utils.py FLUX: save train config (#1049) 2024-11-08 17:15:19 -08:00