LICENSE
README.md
pyproject.toml
src/mflux/__init__.py
src/mflux/generate.py
src/mflux/generate_controlnet.py
src/mflux/generate_in_context.py
src/mflux/save.py
src/mflux/train.py
src/mflux.egg-info/PKG-INFO
src/mflux.egg-info/SOURCES.txt
src/mflux.egg-info/dependency_links.txt
src/mflux.egg-info/entry_points.txt
src/mflux.egg-info/requires.txt
src/mflux.egg-info/top_level.txt
src/mflux/callbacks/__init__.py
src/mflux/callbacks/callback.py
src/mflux/callbacks/callback_registry.py
src/mflux/callbacks/callbacks.py
src/mflux/callbacks/instances/__init__.py
src/mflux/callbacks/instances/canny_saver.py
src/mflux/callbacks/instances/memory_saver.py
src/mflux/callbacks/instances/stepwise_handler.py
src/mflux/community/__init__.py
src/mflux/community/in_context_lora/__init__.py
src/mflux/community/in_context_lora/flux_in_context_lora.py
src/mflux/community/in_context_lora/in_context_loras.py
src/mflux/config/__init__.py
src/mflux/config/config.py
src/mflux/config/model_config.py
src/mflux/config/runtime_config.py
src/mflux/controlnet/__init__.py
src/mflux/controlnet/controlnet_util.py
src/mflux/controlnet/flux_controlnet.py
src/mflux/controlnet/transformer_controlnet.py
src/mflux/controlnet/weight_handler_controlnet.py
src/mflux/dreambooth/__init__.py
src/mflux/dreambooth/dreambooth.py
src/mflux/dreambooth/dreambooth_initializer.py
src/mflux/dreambooth/dataset/__init__.py
src/mflux/dreambooth/dataset/batch.py
src/mflux/dreambooth/dataset/dataset.py
src/mflux/dreambooth/dataset/dreambooth_preprocessing.py
src/mflux/dreambooth/dataset/iterator.py
src/mflux/dreambooth/lora_layers/__init__.py
src/mflux/dreambooth/lora_layers/fused_linear_lora_layer.py
src/mflux/dreambooth/lora_layers/linear_lora_layer.py
src/mflux/dreambooth/lora_layers/lora_layers.py
src/mflux/dreambooth/optimization/__init__.py
src/mflux/dreambooth/optimization/dreambooth_loss.py
src/mflux/dreambooth/optimization/optimizer.py
src/mflux/dreambooth/state/__init__.py
src/mflux/dreambooth/state/training_spec.py
src/mflux/dreambooth/state/training_state.py
src/mflux/dreambooth/state/zip_util.py
src/mflux/dreambooth/statistics/__init__.py
src/mflux/dreambooth/statistics/plotter.py
src/mflux/dreambooth/statistics/statistics.py
src/mflux/error/__init__.py
src/mflux/error/error.py
src/mflux/error/exceptions.py
src/mflux/flux/__init__.py
src/mflux/flux/flux.py
src/mflux/flux/flux_initializer.py
src/mflux/latent_creator/__init__.py
src/mflux/latent_creator/latent_creator.py
src/mflux/models/__init__.py
src/mflux/models/text_encoder/__init__.py
src/mflux/models/text_encoder/prompt_encoder.py
src/mflux/models/text_encoder/clip_encoder/__init__.py
src/mflux/models/text_encoder/clip_encoder/clip_embeddings.py
src/mflux/models/text_encoder/clip_encoder/clip_encoder.py
src/mflux/models/text_encoder/clip_encoder/clip_encoder_layer.py
src/mflux/models/text_encoder/clip_encoder/clip_mlp.py
src/mflux/models/text_encoder/clip_encoder/clip_sdpa_attention.py
src/mflux/models/text_encoder/clip_encoder/clip_text_model.py
src/mflux/models/text_encoder/clip_encoder/encoder_clip.py
src/mflux/models/text_encoder/t5_encoder/__init__.py
src/mflux/models/text_encoder/t5_encoder/t5_attention.py
src/mflux/models/text_encoder/t5_encoder/t5_block.py
src/mflux/models/text_encoder/t5_encoder/t5_dense_relu_dense.py
src/mflux/models/text_encoder/t5_encoder/t5_encoder.py
src/mflux/models/text_encoder/t5_encoder/t5_feed_forward.py
src/mflux/models/text_encoder/t5_encoder/t5_layer_norm.py
src/mflux/models/text_encoder/t5_encoder/t5_self_attention.py
src/mflux/models/transformer/__init__.py
src/mflux/models/transformer/ada_layer_norm_continuous.py
src/mflux/models/transformer/ada_layer_norm_zero.py
src/mflux/models/transformer/ada_layer_norm_zero_single.py
src/mflux/models/transformer/embed_nd.py
src/mflux/models/transformer/feed_forward.py
src/mflux/models/transformer/guidance_embedder.py
src/mflux/models/transformer/joint_attention.py
src/mflux/models/transformer/joint_transformer_block.py
src/mflux/models/transformer/single_block_attention.py
src/mflux/models/transformer/single_transformer_block.py
src/mflux/models/transformer/text_embedder.py
src/mflux/models/transformer/time_text_embed.py
src/mflux/models/transformer/timestep_embedder.py
src/mflux/models/transformer/transformer.py
src/mflux/models/transformer/common/__init__.py
src/mflux/models/transformer/common/attention_utils.py
src/mflux/models/vae/__init__.py
src/mflux/models/vae/vae.py
src/mflux/models/vae/common/__init__.py
src/mflux/models/vae/common/attention.py
src/mflux/models/vae/common/resnet_block_2d.py
src/mflux/models/vae/common/unet_mid_block.py
src/mflux/models/vae/decoder/__init__.py
src/mflux/models/vae/decoder/conv_in.py
src/mflux/models/vae/decoder/conv_norm_out.py
src/mflux/models/vae/decoder/conv_out.py
src/mflux/models/vae/decoder/decoder.py
src/mflux/models/vae/decoder/up_block_1_or_2.py
src/mflux/models/vae/decoder/up_block_3.py
src/mflux/models/vae/decoder/up_block_4.py
src/mflux/models/vae/decoder/up_sampler.py
src/mflux/models/vae/encoder/__init__.py
src/mflux/models/vae/encoder/conv_in.py
src/mflux/models/vae/encoder/conv_norm_out.py
src/mflux/models/vae/encoder/conv_out.py
src/mflux/models/vae/encoder/down_block_1.py
src/mflux/models/vae/encoder/down_block_2.py
src/mflux/models/vae/encoder/down_block_3.py
src/mflux/models/vae/encoder/down_block_4.py
src/mflux/models/vae/encoder/down_sampler.py
src/mflux/models/vae/encoder/encoder.py
src/mflux/post_processing/__init__.py
src/mflux/post_processing/array_util.py
src/mflux/post_processing/generated_image.py
src/mflux/post_processing/image_util.py
src/mflux/tokenizer/__init__.py
src/mflux/tokenizer/clip_tokenizer.py
src/mflux/tokenizer/t5_tokenizer.py
src/mflux/tokenizer/tokenizer_handler.py
src/mflux/ui/__init__.py
src/mflux/ui/defaults.py
src/mflux/ui/cli/__init__.py
src/mflux/ui/cli/parsers.py
src/mflux/weights/__init__.py
src/mflux/weights/lora_converter.py
src/mflux/weights/model_saver.py
src/mflux/weights/quantization_util.py
src/mflux/weights/weight_handler.py
src/mflux/weights/weight_handler_lora.py
src/mflux/weights/weight_handler_lora_huggingface.py
src/mflux/weights/weight_util.py