LICENSE.md
README.md
setup.py
library/__init__.py
library/adafactor_fused.py
library/anima_models.py
library/anima_train_utils.py
library/anima_utils.py
library/attention.py
library/attention_processors.py
library/chroma_models.py
library/config_util.py
library/custom_offloading_utils.py
library/custom_train_functions.py
library/deepspeed_utils.py
library/device_utils.py
library/flux_models.py
library/flux_train_utils.py
library/flux_utils.py
library/fp8_optimization_utils.py
library/huggingface_util.py
library/hunyuan_image_models.py
library/hunyuan_image_modules.py
library/hunyuan_image_text_encoder.py
library/hunyuan_image_utils.py
library/hunyuan_image_vae.py
library/hypernetwork.py
library/jpeg_xl_util.py
library/leco_train_util.py
library/lora_utils.py
library/lpw_stable_diffusion.py
library/lumina_models.py
library/lumina_train_util.py
library/lumina_util.py
library/mask_generator.py
library/model_util.py
library/original_unet.py
library/qwen_image_autoencoder_kl.py
library/safetensors_utils.py
library/sai_model_spec.py
library/sd3_models.py
library/sd3_train_utils.py
library/sd3_utils.py
library/sdxl_lpw_stable_diffusion.py
library/sdxl_model_util.py
library/sdxl_original_control_net.py
library/sdxl_original_unet.py
library/sdxl_train_util.py
library/slicing_vae.py
library/strategy_anima.py
library/strategy_base.py
library/strategy_flux.py
library/strategy_hunyuan_image.py
library/strategy_lumina.py
library/strategy_sd.py
library/strategy_sd3.py
library/strategy_sdxl.py
library/train_util.py
library/utils.py
library.egg-info/PKG-INFO
library.egg-info/SOURCES.txt
library.egg-info/dependency_links.txt
library.egg-info/top_level.txt
library/ipex/__init__.py
library/ipex/attention.py
library/ipex/diffusers.py
library/ipex/hijacks.py
pytorch_lightning/__init__.py
pytorch_lightning/callbacks/__init__.py
pytorch_lightning/callbacks/model_checkpoint.py
tests/test_custom_offloading_utils.py
tests/test_expand_unet_to_inpainting.py
tests/test_fine_tune.py
tests/test_flux_train.py
tests/test_flux_train_network.py
tests/test_lumina_train_network.py
tests/test_mask_generator.py
tests/test_optimizer.py
tests/test_sd3_train.py
tests/test_sd3_train_network.py
tests/test_sdxl_train.py
tests/test_sdxl_train_leco.py
tests/test_sdxl_train_network.py
tests/test_train.py
tests/test_train_leco.py
tests/test_train_network.py
tests/test_train_textual_inversion.py
tests/test_validation.py