├── .github └── workflows │ └── release.yml ├── .gitignore ├── .pre-commit-config.yaml ├── CHANGELOG.md ├── LICENSE ├── Makefile ├── README.md ├── _typos.toml ├── pyproject.toml ├── src └── mflux │ ├── __init__.py │ ├── assets │ ├── catvton_example.jpg │ ├── catvton_how_it_works.jpg │ ├── comparison1.jpg │ ├── comparison2.jpg │ ├── comparison3.jpg │ ├── comparison4.jpg │ ├── comparison5.jpg │ ├── comparison6.jpg │ ├── concept_example_1.jpg │ ├── concept_example_2.jpg │ ├── concept_example_3.jpg │ ├── controlnet1.jpg │ ├── controlnet2.jpg │ ├── depth_example.jpg │ ├── dreambooth.jpg │ ├── fibo_example.jpg │ ├── fibo_inspire_example.jpg │ ├── fibo_refine_example.jpg │ ├── ic_edit_example.jpg │ ├── ic_edit_how_it_works.jpg │ ├── img2img.jpg │ ├── in_context_example.jpg │ ├── in_context_how_it_works.jpg │ ├── inpaint_example.jpg │ ├── kontext_example.jpg │ ├── kontext_sequential_edit_example.jpg │ ├── krea_dev_example.jpg │ ├── logo.png │ ├── lora1.jpg │ ├── lora2.jpg │ ├── lora3.jpg │ ├── outpaint_example.jpg │ ├── qwen_edit_example.jpg │ ├── qwen_image_example.jpg │ ├── redux_example.jpg │ └── upscale_example.jpg │ ├── callbacks │ ├── __init__.py │ ├── callback.py │ ├── callback_manager.py │ ├── callback_registry.py │ ├── callbacks.py │ └── instances │ │ ├── __init__.py │ │ ├── battery_saver.py │ │ ├── canny_saver.py │ │ ├── depth_saver.py │ │ ├── memory_saver.py │ │ └── stepwise_handler.py │ ├── concept.py │ ├── concept_from_image.py │ ├── config │ ├── __init__.py │ ├── config.py │ ├── model_config.py │ └── runtime_config.py │ ├── generate.py │ ├── generate_controlnet.py │ ├── generate_depth.py │ ├── generate_fibo.py │ ├── generate_fill.py │ ├── generate_in_context_catvton.py │ ├── generate_in_context_dev.py │ ├── generate_in_context_edit.py │ ├── generate_kontext.py │ ├── generate_qwen.py │ ├── generate_qwen_edit.py │ ├── generate_redux.py │ ├── info.py │ ├── inspire_fibo.py │ ├── lora_library.py │ ├── models │ ├── __init__.py │ ├── common │ │ ├── __init__.py │ │ ├── latent_creator │ │ │ ├── __init__.py │ │ │ └── latent_creator.py │ │ ├── lora │ │ │ ├── __init__.py │ │ │ ├── download │ │ │ │ ├── __init__.py │ │ │ │ ├── lora_huggingface_downloader.py │ │ │ │ └── lora_library.py │ │ │ ├── layer │ │ │ │ ├── __init__.py │ │ │ │ ├── fused_linear_lora_layer.py │ │ │ │ └── linear_lora_layer.py │ │ │ └── mapping │ │ │ │ ├── __init__.py │ │ │ │ ├── lora_loader.py │ │ │ │ └── lora_mapping.py │ │ ├── quantization │ │ │ ├── __init__.py │ │ │ └── quantization_util.py │ │ ├── schedulers │ │ │ ├── __init__.py │ │ │ ├── base_scheduler.py │ │ │ ├── flow_match_euler_discrete_scheduler.py │ │ │ └── linear_scheduler.py │ │ └── weights │ │ │ ├── __init__.py │ │ │ ├── mapping │ │ │ ├── __init__.py │ │ │ ├── weight_mapper.py │ │ │ └── weight_mapping.py │ │ │ └── model_saver.py │ ├── depth_pro │ │ ├── __init__.py │ │ ├── depth_pro.py │ │ ├── depth_pro_initializer.py │ │ ├── model │ │ │ ├── __init__.py │ │ │ ├── conv_utils.py │ │ │ ├── depth_pro_encoder.py │ │ │ ├── depth_pro_model.py │ │ │ ├── depth_pro_util.py │ │ │ ├── dino_v2 │ │ │ │ ├── __init__.py │ │ │ │ ├── attention.py │ │ │ │ ├── dino_vision_transformer.py │ │ │ │ ├── layer_scale.py │ │ │ │ ├── mlp.py │ │ │ │ ├── patch_embed.py │ │ │ │ └── transformer_block.py │ │ │ ├── feature_fusion_block_2d.py │ │ │ ├── fov_head.py │ │ │ ├── multires_conv_decoder.py │ │ │ ├── residual_block.py │ │ │ └── upsample_block.py │ │ └── weights │ │ │ ├── __init__.py │ │ │ └── weight_handler_depth_pro.py │ ├── fibo │ │ ├── __init__.py │ │ ├── fibo_initializer.py │ │ ├── latent_creator │ │ │ ├── __init__.py │ │ │ └── fibo_latent_creator.py │ │ ├── model │ │ │ ├── __init__.py │ │ │ ├── fibo_text_encoder │ │ │ │ ├── __init__.py │ │ │ │ ├── prompt_encoder.py │ │ │ │ ├── smol_lm3_3b_attention.py │ │ │ │ ├── smol_lm3_3b_encoder_layer.py │ │ │ │ ├── smol_lm3_3b_mlp.py │ │ │ │ ├── smol_lm3_3b_rms_norm.py │ │ │ │ ├── smol_lm3_3b_rope.py │ │ │ │ └── smol_lm3_3b_text_encoder.py │ │ │ ├── fibo_transformer │ │ │ │ ├── __init__.py │ │ │ │ ├── bria_fibo_timesteps.py │ │ │ │ ├── feed_forward.py │ │ │ │ ├── fibo_ada_layer_norm_zero.py │ │ │ │ ├── fibo_embed_nd.py │ │ │ │ ├── fibo_gelu.py │ │ │ │ ├── fibo_joint_attention.py │ │ │ │ ├── fibo_single_attention.py │ │ │ │ ├── joint_transformer_block.py │ │ │ │ ├── single_transformer_block.py │ │ │ │ ├── text_projection.py │ │ │ │ ├── time_embed.py │ │ │ │ └── transformer.py │ │ │ └── fibo_vae │ │ │ │ ├── __init__.py │ │ │ │ ├── common │ │ │ │ ├── __init__.py │ │ │ │ ├── wan_2_2_attention_block.py │ │ │ │ ├── wan_2_2_causal_conv_3d.py │ │ │ │ ├── wan_2_2_mid_block.py │ │ │ │ ├── wan_2_2_resample.py │ │ │ │ ├── wan_2_2_residual_block.py │ │ │ │ └── wan_2_2_rms_norm.py │ │ │ │ ├── decoder │ │ │ │ ├── __init__.py │ │ │ │ ├── wan_2_2_decoder_3d.py │ │ │ │ ├── wan_2_2_dup_up_3d.py │ │ │ │ ├── wan_2_2_residual_up_block.py │ │ │ │ └── wan_2_2_up_block.py │ │ │ │ ├── encoder │ │ │ │ ├── __init__.py │ │ │ │ ├── wan_2_2_avg_down_3d.py │ │ │ │ ├── wan_2_2_down_block.py │ │ │ │ └── wan_2_2_encoder_3d.py │ │ │ │ └── wan_2_2_vae.py │ │ ├── tokenizer │ │ │ ├── __init__.py │ │ │ ├── fibo_tokenizer.py │ │ │ ├── qwen2vl_image_processor.py │ │ │ ├── qwen2vl_processor.py │ │ │ └── smol_lm3_3b_tokenizer.py │ │ ├── variants │ │ │ ├── __init__.py │ │ │ └── txt2img │ │ │ │ ├── __init__.py │ │ │ │ └── fibo.py │ │ └── weights │ │ │ ├── __init__.py │ │ │ ├── fibo_weight_handler.py │ │ │ ├── fibo_weight_mapping.py │ │ │ └── fibo_weight_util.py │ ├── fibo_vlm │ │ ├── __init__.py │ │ ├── fibo_vlm_initializer.py │ │ ├── model │ │ │ ├── __init__.py │ │ │ ├── fibo_vlm.py │ │ │ ├── qwen3_vl_attention.py │ │ │ ├── qwen3_vl_decoder.py │ │ │ ├── qwen3_vl_decoder_layer.py │ │ │ ├── qwen3_vl_mlp.py │ │ │ ├── qwen3_vl_rms_norm.py │ │ │ ├── qwen3_vl_rope.py │ │ │ ├── qwen3_vl_util.py │ │ │ ├── qwen3_vl_vision_attention.py │ │ │ ├── qwen3_vl_vision_block.py │ │ │ ├── qwen3_vl_vision_mlp.py │ │ │ ├── qwen3_vl_vision_model.py │ │ │ ├── qwen3_vl_vision_patch_embed.py │ │ │ ├── qwen3_vl_vision_patch_merger.py │ │ │ └── qwen3_vl_vision_rotary_embedding.py │ │ └── weights │ │ │ ├── __init__.py │ │ │ ├── fibo_vlm_weight_handler.py │ │ │ └── fibo_vlm_weight_mapping.py │ ├── flux │ │ ├── __init__.py │ │ ├── flux_initializer.py │ │ ├── latent_creator │ │ │ ├── __init__.py │ │ │ └── flux_latent_creator.py │ │ ├── model │ │ │ ├── __init__.py │ │ │ ├── flux_text_encoder │ │ │ │ ├── __init__.py │ │ │ │ ├── clip_encoder │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── clip_embeddings.py │ │ │ │ │ ├── clip_encoder.py │ │ │ │ │ ├── clip_encoder_layer.py │ │ │ │ │ ├── clip_mlp.py │ │ │ │ │ ├── clip_sdpa_attention.py │ │ │ │ │ ├── clip_text_model.py │ │ │ │ │ └── encoder_clip.py │ │ │ │ ├── prompt_encoder.py │ │ │ │ └── t5_encoder │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── t5_attention.py │ │ │ │ │ ├── t5_block.py │ │ │ │ │ ├── t5_dense_relu_dense.py │ │ │ │ │ ├── t5_encoder.py │ │ │ │ │ ├── t5_feed_forward.py │ │ │ │ │ ├── t5_layer_norm.py │ │ │ │ │ └── t5_self_attention.py │ │ │ ├── flux_transformer │ │ │ │ ├── __init__.py │ │ │ │ ├── ada_layer_norm_continuous.py │ │ │ │ ├── ada_layer_norm_zero.py │ │ │ │ ├── ada_layer_norm_zero_single.py │ │ │ │ ├── common │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── attention_utils.py │ │ │ │ ├── embed_nd.py │ │ │ │ ├── feed_forward.py │ │ │ │ ├── guidance_embedder.py │ │ │ │ ├── joint_attention.py │ │ │ │ ├── joint_transformer_block.py │ │ │ │ ├── single_block_attention.py │ │ │ │ ├── single_transformer_block.py │ │ │ │ ├── text_embedder.py │ │ │ │ ├── time_text_embed.py │ │ │ │ ├── timestep_embedder.py │ │ │ │ └── transformer.py │ │ │ ├── flux_vae │ │ │ │ ├── __init__.py │ │ │ │ ├── common │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── attention.py │ │ │ │ │ ├── resnet_block_2d.py │ │ │ │ │ └── unet_mid_block.py │ │ │ │ ├── decoder │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── conv_in.py │ │ │ │ │ ├── conv_norm_out.py │ │ │ │ │ ├── conv_out.py │ │ │ │ │ ├── decoder.py │ │ │ │ │ ├── up_block_1_or_2.py │ │ │ │ │ ├── up_block_3.py │ │ │ │ │ ├── up_block_4.py │ │ │ │ │ └── up_sampler.py │ │ │ │ ├── encoder │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── conv_in.py │ │ │ │ │ ├── conv_norm_out.py │ │ │ │ │ ├── conv_out.py │ │ │ │ │ ├── down_block_1.py │ │ │ │ │ ├── down_block_2.py │ │ │ │ │ ├── down_block_3.py │ │ │ │ │ ├── down_block_4.py │ │ │ │ │ ├── down_sampler.py │ │ │ │ │ └── encoder.py │ │ │ │ └── vae.py │ │ │ ├── redux_encoder │ │ │ │ ├── __init__.py │ │ │ │ └── redux_encoder.py │ │ │ └── siglip_vision_transformer │ │ │ │ ├── __init__.py │ │ │ │ ├── siglip_encoder.py │ │ │ │ ├── siglip_encoder_layer.py │ │ │ │ ├── siglip_mlp.py │ │ │ │ ├── siglip_multi_head_attention_pooling_head.py │ │ │ │ ├── siglip_sdpa_attention.py │ │ │ │ ├── siglip_vision_embeddings.py │ │ │ │ └── siglip_vision_transformer.py │ │ ├── tokenizer │ │ │ ├── __init__.py │ │ │ ├── clip_tokenizer.py │ │ │ ├── t5_tokenizer.py │ │ │ └── tokenizer_handler.py │ │ ├── variants │ │ │ ├── __init__.py │ │ │ ├── concept_attention │ │ │ │ ├── __init__.py │ │ │ │ ├── attention_data.py │ │ │ │ ├── concept_util.py │ │ │ │ ├── flux_concept.py │ │ │ │ ├── flux_concept_from_image.py │ │ │ │ ├── joint_attention_concept.py │ │ │ │ ├── joint_transformer_block_concept.py │ │ │ │ └── transformer_concept.py │ │ │ ├── controlnet │ │ │ │ ├── __init__.py │ │ │ │ ├── controlnet_util.py │ │ │ │ ├── flux_controlnet.py │ │ │ │ ├── transformer_controlnet.py │ │ │ │ └── weight_handler_controlnet.py │ │ │ ├── depth │ │ │ │ ├── __init__.py │ │ │ │ ├── depth_util.py │ │ │ │ └── flux_depth.py │ │ │ ├── dreambooth │ │ │ │ ├── __init__.py │ │ │ │ ├── _example │ │ │ │ │ ├── images │ │ │ │ │ │ ├── 01.jpeg │ │ │ │ │ │ ├── 02.jpeg │ │ │ │ │ │ ├── 03.jpeg │ │ │ │ │ │ ├── 04.jpeg │ │ │ │ │ │ └── 05.jpeg │ │ │ │ │ └── train.json │ │ │ │ ├── dataset │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── batch.py │ │ │ │ │ ├── dataset.py │ │ │ │ │ ├── dreambooth_preprocessing.py │ │ │ │ │ └── iterator.py │ │ │ │ ├── dreambooth.py │ │ │ │ ├── dreambooth_initializer.py │ │ │ │ ├── lora_layers │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── lora_layers.py │ │ │ │ ├── optimization │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── _loss_derivation │ │ │ │ │ │ ├── dreambooth_loss.pdf │ │ │ │ │ │ └── dreambooth_loss.tex │ │ │ │ │ ├── dreambooth_loss.py │ │ │ │ │ └── optimizer.py │ │ │ │ ├── state │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── training_spec.py │ │ │ │ │ ├── training_state.py │ │ │ │ │ └── zip_util.py │ │ │ │ └── statistics │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── plotter.py │ │ │ │ │ └── statistics.py │ │ │ ├── fill │ │ │ │ ├── __init__.py │ │ │ │ ├── flux_fill.py │ │ │ │ └── mask_util.py │ │ │ ├── in_context │ │ │ │ ├── __init__.py │ │ │ │ ├── flux_in_context_dev.py │ │ │ │ ├── flux_in_context_fill.py │ │ │ │ └── utils │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── in_context_loras.py │ │ │ │ │ └── in_context_mask_util.py │ │ │ ├── kontext │ │ │ │ ├── __init__.py │ │ │ │ ├── flux_kontext.py │ │ │ │ └── utils │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── kontext_util.py │ │ │ ├── redux │ │ │ │ ├── __init__.py │ │ │ │ ├── flux_redux.py │ │ │ │ ├── redux_util.py │ │ │ │ └── weight_handler_redux.py │ │ │ └── txt2img │ │ │ │ ├── __init__.py │ │ │ │ └── flux.py │ │ └── weights │ │ │ ├── __init__.py │ │ │ ├── lora_converter.py │ │ │ ├── weight_handler.py │ │ │ ├── weight_handler_lora.py │ │ │ └── weight_util.py │ └── qwen │ │ ├── __init__.py │ │ ├── latent_creator │ │ ├── __init__.py │ │ └── qwen_latent_creator.py │ │ ├── model │ │ ├── __init__.py │ │ ├── qwen_text_encoder │ │ │ ├── __init__.py │ │ │ ├── qwen_attention.py │ │ │ ├── qwen_encoder.py │ │ │ ├── qwen_encoder_layer.py │ │ │ ├── qwen_mlp.py │ │ │ ├── qwen_patch_merger.py │ │ │ ├── qwen_prompt_encoder.py │ │ │ ├── qwen_rms_norm.py │ │ │ ├── qwen_rope.py │ │ │ ├── qwen_text_encoder.py │ │ │ ├── qwen_vision_attention.py │ │ │ ├── qwen_vision_block.py │ │ │ ├── qwen_vision_language_encoder.py │ │ │ ├── qwen_vision_language_prompt_encoder.py │ │ │ ├── qwen_vision_mlp.py │ │ │ ├── qwen_vision_patch_embed.py │ │ │ ├── qwen_vision_rotary_embedding.py │ │ │ └── qwen_vision_transformer.py │ │ ├── qwen_transformer │ │ │ ├── __init__.py │ │ │ ├── qwen_attention.py │ │ │ ├── qwen_feed_forward.py │ │ │ ├── qwen_layer_norm.py │ │ │ ├── qwen_rope.py │ │ │ ├── qwen_time_text_embed.py │ │ │ ├── qwen_timestep_embedding.py │ │ │ ├── qwen_timesteps.py │ │ │ ├── qwen_transformer.py │ │ │ ├── qwen_transformer_block.py │ │ │ └── qwen_transformer_rms_norm.py │ │ └── qwen_vae │ │ │ ├── __init__.py │ │ │ ├── qwen_image_attention_block_3d.py │ │ │ ├── qwen_image_causal_conv_3d.py │ │ │ ├── qwen_image_decoder_3d.py │ │ │ ├── qwen_image_down_block_3d.py │ │ │ ├── qwen_image_encoder_3d.py │ │ │ ├── qwen_image_mid_block_3d.py │ │ │ ├── qwen_image_res_block_3d.py │ │ │ ├── qwen_image_resample_3d.py │ │ │ ├── qwen_image_rms_norm.py │ │ │ ├── qwen_image_up_block_3d.py │ │ │ └── qwen_vae.py │ │ ├── qwen_edit_initializer.py │ │ ├── qwen_initializer.py │ │ ├── tokenizer │ │ ├── __init__.py │ │ ├── qwen_image_processor.py │ │ ├── qwen_tokenizer.py │ │ ├── qwen_tokenizer_handler.py │ │ ├── qwen_vision_language_processor.py │ │ └── qwen_vision_language_tokenizer.py │ │ ├── variants │ │ ├── __init__.py │ │ ├── edit │ │ │ ├── __init__.py │ │ │ ├── qwen_image_edit.py │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── qwen_edit_util.py │ │ │ │ └── qwen_prompt_rewriter.py │ │ └── txt2img │ │ │ ├── __init__.py │ │ │ └── qwen_image.py │ │ └── weights │ │ ├── __init__.py │ │ ├── qwen_lora_mapping.py │ │ ├── qwen_weight_handler.py │ │ ├── qwen_weight_mapping.py │ │ └── qwen_weight_util.py │ ├── refine_fibo.py │ ├── release │ ├── __init__.py │ ├── changelog_parser.py │ ├── git_operations.py │ ├── github_api.py │ ├── pypi_publisher.py │ ├── release_manager.py │ ├── release_script.py │ └── release_validator.py │ ├── save.py │ ├── save_depth.py │ ├── train.py │ ├── ui │ ├── __init__.py │ ├── box_values.py │ ├── cli │ │ ├── __init__.py │ │ ├── completions │ │ │ ├── README.md │ │ │ ├── __init__.py │ │ │ ├── generator.py │ │ │ └── install.py │ │ └── parsers.py │ ├── defaults.py │ ├── prompt_utils.py │ └── scale_factor.py │ ├── upscale.py │ └── utils │ ├── __init__.py │ ├── array_util.py │ ├── download.py │ ├── exceptions.py │ ├── generated_image.py │ ├── image_compare.py │ ├── image_util.py │ ├── metadata_builder.py │ ├── metadata_reader.py │ └── version_util.py ├── tests ├── __init__.py ├── arg_parser │ ├── __init__.py │ ├── test_cli_argparser.py │ ├── test_lora_library_integration.py │ ├── test_stdin_prompt.py │ └── test_upscale_argparser.py ├── callbacks │ ├── __init__.py │ └── test_battery_saver.py ├── depth │ ├── __init__.py │ └── test_depth_pro.py ├── dreambooth │ ├── __init__.py │ ├── config │ │ └── train.json │ ├── test_dreambooth_dataset_iterator.py │ ├── test_resume_training.py │ └── test_train_and_load_weights.py ├── image_generation │ ├── __init__.py │ ├── helpers │ │ ├── __init__.py │ │ ├── image_generation_concept_test_helper.py │ │ ├── image_generation_controlnet_test_helper.py │ │ ├── image_generation_depth_test_helper.py │ │ ├── image_generation_edit_test_helper.py │ │ ├── image_generation_fibo_test_helper.py │ │ ├── image_generation_fill_test_helper.py │ │ ├── image_generation_ic_edit_test_helper.py │ │ ├── image_generation_in_context_test_helper.py │ │ ├── image_generation_kontext_test_helper.py │ │ ├── image_generation_redux_test_helper.py │ │ └── image_generation_test_helper.py │ ├── test_generate_concept.py │ ├── test_generate_image.py │ ├── test_generate_image_controlnet.py │ ├── test_generate_image_depth.py │ ├── test_generate_image_fibo.py │ ├── test_generate_image_fill.py │ ├── test_generate_image_in_context.py │ ├── test_generate_image_kontext.py │ ├── test_generate_image_qwen_image.py │ ├── test_generate_image_qwen_image_edit.py │ ├── test_generate_image_redux.py │ ├── test_image_util.py │ ├── test_stdin_prompt_integration.py │ └── test_upscale_dimensions.py ├── metadata │ ├── __init__.py │ └── test_metadata.py ├── model_config │ ├── __init__.py │ └── test_model_config.py ├── model_saving │ ├── __init__.py │ ├── test_model_saving.py │ └── test_model_saving_lora.py ├── resources │ ├── .gitignore │ ├── FLUX-dev-lora-MiaoKa-Yarn-World.safetensors │ ├── Flux_-_Renaissance_art_style.safetensors │ ├── controlnet_reference.png │ ├── ic_lora_reference_logo.png │ ├── low_res.jpg │ ├── mask.png │ ├── reference_concept_from_image_schnell_heatmap.png │ ├── reference_concept_schnell_heatmap.png │ ├── reference_controlnet_dev.png │ ├── reference_controlnet_dev_lora.png │ ├── reference_controlnet_dev_lora_depth.png │ ├── reference_controlnet_schnell.png │ ├── reference_depth_dev_from_image.png │ ├── reference_dev.png │ ├── reference_dev_image_to_image.png │ ├── reference_dev_image_to_image_fill.png │ ├── reference_dev_kontext.png │ ├── reference_dev_lora.png │ ├── reference_dev_lora_multiple.png │ ├── reference_fibo.png │ ├── reference_fibo_white_owl.png │ ├── reference_ic_edit.png │ ├── reference_ic_lora_identity.png │ ├── reference_qwen_edit.png │ ├── reference_qwen_edit_multiple_images.png │ ├── reference_qwen_img2img.png │ ├── reference_qwen_lora.png │ ├── reference_qwen_txt2img.png │ ├── reference_redux_dev.png │ ├── reference_schnell.png │ ├── reference_upscaled.png │ ├── shirt.jpg │ └── skyscrapers.jpg ├── schedulers │ ├── __init__.py │ ├── test_base_scheduler.py │ ├── test_linear_scheduler.py │ └── test_scheduler_lookup.py ├── ui │ ├── __init__.py │ └── test_scale_factor.py ├── vlm │ ├── __init__.py │ └── test_fibo_vlm.py └── weights │ ├── __init__.py │ └── test_lora_library.py ├── tools ├── create_outpaint_image_canvas_and_mask.py ├── inpaint_mask_tool.py └── rename_images.py └── uv.lock /.github/workflows/release.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/.github/workflows/release.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/README.md -------------------------------------------------------------------------------- /_typos.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/_typos.toml -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/pyproject.toml -------------------------------------------------------------------------------- /src/mflux/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/__init__.py -------------------------------------------------------------------------------- /src/mflux/assets/catvton_example.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/catvton_example.jpg -------------------------------------------------------------------------------- /src/mflux/assets/catvton_how_it_works.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/catvton_how_it_works.jpg -------------------------------------------------------------------------------- /src/mflux/assets/comparison1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/comparison1.jpg -------------------------------------------------------------------------------- /src/mflux/assets/comparison2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/comparison2.jpg -------------------------------------------------------------------------------- /src/mflux/assets/comparison3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/comparison3.jpg -------------------------------------------------------------------------------- /src/mflux/assets/comparison4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/comparison4.jpg -------------------------------------------------------------------------------- /src/mflux/assets/comparison5.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/comparison5.jpg -------------------------------------------------------------------------------- /src/mflux/assets/comparison6.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/comparison6.jpg -------------------------------------------------------------------------------- /src/mflux/assets/concept_example_1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/concept_example_1.jpg -------------------------------------------------------------------------------- /src/mflux/assets/concept_example_2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/concept_example_2.jpg -------------------------------------------------------------------------------- /src/mflux/assets/concept_example_3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/concept_example_3.jpg -------------------------------------------------------------------------------- /src/mflux/assets/controlnet1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/controlnet1.jpg -------------------------------------------------------------------------------- /src/mflux/assets/controlnet2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/controlnet2.jpg -------------------------------------------------------------------------------- /src/mflux/assets/depth_example.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/depth_example.jpg -------------------------------------------------------------------------------- /src/mflux/assets/dreambooth.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/dreambooth.jpg -------------------------------------------------------------------------------- /src/mflux/assets/fibo_example.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/fibo_example.jpg -------------------------------------------------------------------------------- /src/mflux/assets/fibo_inspire_example.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/fibo_inspire_example.jpg -------------------------------------------------------------------------------- /src/mflux/assets/fibo_refine_example.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/fibo_refine_example.jpg -------------------------------------------------------------------------------- /src/mflux/assets/ic_edit_example.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/ic_edit_example.jpg -------------------------------------------------------------------------------- /src/mflux/assets/ic_edit_how_it_works.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/ic_edit_how_it_works.jpg -------------------------------------------------------------------------------- /src/mflux/assets/img2img.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/img2img.jpg -------------------------------------------------------------------------------- /src/mflux/assets/in_context_example.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/in_context_example.jpg -------------------------------------------------------------------------------- /src/mflux/assets/in_context_how_it_works.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/in_context_how_it_works.jpg -------------------------------------------------------------------------------- /src/mflux/assets/inpaint_example.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/inpaint_example.jpg -------------------------------------------------------------------------------- /src/mflux/assets/kontext_example.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/kontext_example.jpg -------------------------------------------------------------------------------- /src/mflux/assets/kontext_sequential_edit_example.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/kontext_sequential_edit_example.jpg -------------------------------------------------------------------------------- /src/mflux/assets/krea_dev_example.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/krea_dev_example.jpg -------------------------------------------------------------------------------- /src/mflux/assets/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/logo.png -------------------------------------------------------------------------------- /src/mflux/assets/lora1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/lora1.jpg -------------------------------------------------------------------------------- /src/mflux/assets/lora2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/lora2.jpg -------------------------------------------------------------------------------- /src/mflux/assets/lora3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/lora3.jpg -------------------------------------------------------------------------------- /src/mflux/assets/outpaint_example.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/outpaint_example.jpg -------------------------------------------------------------------------------- /src/mflux/assets/qwen_edit_example.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/qwen_edit_example.jpg -------------------------------------------------------------------------------- /src/mflux/assets/qwen_image_example.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/qwen_image_example.jpg -------------------------------------------------------------------------------- /src/mflux/assets/redux_example.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/redux_example.jpg -------------------------------------------------------------------------------- /src/mflux/assets/upscale_example.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/assets/upscale_example.jpg -------------------------------------------------------------------------------- /src/mflux/callbacks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/callbacks/callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/callbacks/callback.py -------------------------------------------------------------------------------- /src/mflux/callbacks/callback_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/callbacks/callback_manager.py -------------------------------------------------------------------------------- /src/mflux/callbacks/callback_registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/callbacks/callback_registry.py -------------------------------------------------------------------------------- /src/mflux/callbacks/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/callbacks/callbacks.py -------------------------------------------------------------------------------- /src/mflux/callbacks/instances/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/callbacks/instances/battery_saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/callbacks/instances/battery_saver.py -------------------------------------------------------------------------------- /src/mflux/callbacks/instances/canny_saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/callbacks/instances/canny_saver.py -------------------------------------------------------------------------------- /src/mflux/callbacks/instances/depth_saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/callbacks/instances/depth_saver.py -------------------------------------------------------------------------------- /src/mflux/callbacks/instances/memory_saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/callbacks/instances/memory_saver.py -------------------------------------------------------------------------------- /src/mflux/callbacks/instances/stepwise_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/callbacks/instances/stepwise_handler.py -------------------------------------------------------------------------------- /src/mflux/concept.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/concept.py -------------------------------------------------------------------------------- /src/mflux/concept_from_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/concept_from_image.py -------------------------------------------------------------------------------- /src/mflux/config/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/config/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/config/config.py -------------------------------------------------------------------------------- /src/mflux/config/model_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/config/model_config.py -------------------------------------------------------------------------------- /src/mflux/config/runtime_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/config/runtime_config.py -------------------------------------------------------------------------------- /src/mflux/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/generate.py -------------------------------------------------------------------------------- /src/mflux/generate_controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/generate_controlnet.py -------------------------------------------------------------------------------- /src/mflux/generate_depth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/generate_depth.py -------------------------------------------------------------------------------- /src/mflux/generate_fibo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/generate_fibo.py -------------------------------------------------------------------------------- /src/mflux/generate_fill.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/generate_fill.py -------------------------------------------------------------------------------- /src/mflux/generate_in_context_catvton.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/generate_in_context_catvton.py -------------------------------------------------------------------------------- /src/mflux/generate_in_context_dev.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/generate_in_context_dev.py -------------------------------------------------------------------------------- /src/mflux/generate_in_context_edit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/generate_in_context_edit.py -------------------------------------------------------------------------------- /src/mflux/generate_kontext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/generate_kontext.py -------------------------------------------------------------------------------- /src/mflux/generate_qwen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/generate_qwen.py -------------------------------------------------------------------------------- /src/mflux/generate_qwen_edit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/generate_qwen_edit.py -------------------------------------------------------------------------------- /src/mflux/generate_redux.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/generate_redux.py -------------------------------------------------------------------------------- /src/mflux/info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/info.py -------------------------------------------------------------------------------- /src/mflux/inspire_fibo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/inspire_fibo.py -------------------------------------------------------------------------------- /src/mflux/lora_library.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/lora_library.py -------------------------------------------------------------------------------- /src/mflux/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/common/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/common/latent_creator/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/common/latent_creator/latent_creator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/common/latent_creator/latent_creator.py -------------------------------------------------------------------------------- /src/mflux/models/common/lora/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/common/lora/download/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/common/lora/download/lora_huggingface_downloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/common/lora/download/lora_huggingface_downloader.py -------------------------------------------------------------------------------- /src/mflux/models/common/lora/download/lora_library.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/common/lora/download/lora_library.py -------------------------------------------------------------------------------- /src/mflux/models/common/lora/layer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/common/lora/layer/fused_linear_lora_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/common/lora/layer/fused_linear_lora_layer.py -------------------------------------------------------------------------------- /src/mflux/models/common/lora/layer/linear_lora_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/common/lora/layer/linear_lora_layer.py -------------------------------------------------------------------------------- /src/mflux/models/common/lora/mapping/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/common/lora/mapping/lora_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/common/lora/mapping/lora_loader.py -------------------------------------------------------------------------------- /src/mflux/models/common/lora/mapping/lora_mapping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/common/lora/mapping/lora_mapping.py -------------------------------------------------------------------------------- /src/mflux/models/common/quantization/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/common/quantization/quantization_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/common/quantization/quantization_util.py -------------------------------------------------------------------------------- /src/mflux/models/common/schedulers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/common/schedulers/__init__.py -------------------------------------------------------------------------------- /src/mflux/models/common/schedulers/base_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/common/schedulers/base_scheduler.py -------------------------------------------------------------------------------- /src/mflux/models/common/schedulers/flow_match_euler_discrete_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/common/schedulers/flow_match_euler_discrete_scheduler.py -------------------------------------------------------------------------------- /src/mflux/models/common/schedulers/linear_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/common/schedulers/linear_scheduler.py -------------------------------------------------------------------------------- /src/mflux/models/common/weights/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/common/weights/__init__.py -------------------------------------------------------------------------------- /src/mflux/models/common/weights/mapping/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/common/weights/mapping/__init__.py -------------------------------------------------------------------------------- /src/mflux/models/common/weights/mapping/weight_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/common/weights/mapping/weight_mapper.py -------------------------------------------------------------------------------- /src/mflux/models/common/weights/mapping/weight_mapping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/common/weights/mapping/weight_mapping.py -------------------------------------------------------------------------------- /src/mflux/models/common/weights/model_saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/common/weights/model_saver.py -------------------------------------------------------------------------------- /src/mflux/models/depth_pro/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/depth_pro/depth_pro.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/depth_pro/depth_pro.py -------------------------------------------------------------------------------- /src/mflux/models/depth_pro/depth_pro_initializer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/depth_pro/depth_pro_initializer.py -------------------------------------------------------------------------------- /src/mflux/models/depth_pro/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/depth_pro/model/conv_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/depth_pro/model/conv_utils.py -------------------------------------------------------------------------------- /src/mflux/models/depth_pro/model/depth_pro_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/depth_pro/model/depth_pro_encoder.py -------------------------------------------------------------------------------- /src/mflux/models/depth_pro/model/depth_pro_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/depth_pro/model/depth_pro_model.py -------------------------------------------------------------------------------- /src/mflux/models/depth_pro/model/depth_pro_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/depth_pro/model/depth_pro_util.py -------------------------------------------------------------------------------- /src/mflux/models/depth_pro/model/dino_v2/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/depth_pro/model/dino_v2/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/depth_pro/model/dino_v2/attention.py -------------------------------------------------------------------------------- /src/mflux/models/depth_pro/model/dino_v2/dino_vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/depth_pro/model/dino_v2/dino_vision_transformer.py -------------------------------------------------------------------------------- /src/mflux/models/depth_pro/model/dino_v2/layer_scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/depth_pro/model/dino_v2/layer_scale.py -------------------------------------------------------------------------------- /src/mflux/models/depth_pro/model/dino_v2/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/depth_pro/model/dino_v2/mlp.py -------------------------------------------------------------------------------- /src/mflux/models/depth_pro/model/dino_v2/patch_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/depth_pro/model/dino_v2/patch_embed.py -------------------------------------------------------------------------------- /src/mflux/models/depth_pro/model/dino_v2/transformer_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/depth_pro/model/dino_v2/transformer_block.py -------------------------------------------------------------------------------- /src/mflux/models/depth_pro/model/feature_fusion_block_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/depth_pro/model/feature_fusion_block_2d.py -------------------------------------------------------------------------------- /src/mflux/models/depth_pro/model/fov_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/depth_pro/model/fov_head.py -------------------------------------------------------------------------------- /src/mflux/models/depth_pro/model/multires_conv_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/depth_pro/model/multires_conv_decoder.py -------------------------------------------------------------------------------- /src/mflux/models/depth_pro/model/residual_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/depth_pro/model/residual_block.py -------------------------------------------------------------------------------- /src/mflux/models/depth_pro/model/upsample_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/depth_pro/model/upsample_block.py -------------------------------------------------------------------------------- /src/mflux/models/depth_pro/weights/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/depth_pro/weights/weight_handler_depth_pro.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/depth_pro/weights/weight_handler_depth_pro.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/fibo/fibo_initializer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/fibo_initializer.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/latent_creator/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/fibo/latent_creator/fibo_latent_creator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/latent_creator/fibo_latent_creator.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/__init__.py: -------------------------------------------------------------------------------- 1 | """FIBO model components.""" 2 | -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_text_encoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_text_encoder/__init__.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_text_encoder/prompt_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_text_encoder/prompt_encoder.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_text_encoder/smol_lm3_3b_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_text_encoder/smol_lm3_3b_attention.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_text_encoder/smol_lm3_3b_encoder_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_text_encoder/smol_lm3_3b_encoder_layer.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_text_encoder/smol_lm3_3b_mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_text_encoder/smol_lm3_3b_mlp.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_text_encoder/smol_lm3_3b_rms_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_text_encoder/smol_lm3_3b_rms_norm.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_text_encoder/smol_lm3_3b_rope.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_text_encoder/smol_lm3_3b_rope.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_text_encoder/smol_lm3_3b_text_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_text_encoder/smol_lm3_3b_text_encoder.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_transformer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_transformer/__init__.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_transformer/bria_fibo_timesteps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_transformer/bria_fibo_timesteps.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_transformer/feed_forward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_transformer/feed_forward.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_transformer/fibo_ada_layer_norm_zero.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_transformer/fibo_ada_layer_norm_zero.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_transformer/fibo_embed_nd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_transformer/fibo_embed_nd.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_transformer/fibo_gelu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_transformer/fibo_gelu.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_transformer/fibo_joint_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_transformer/fibo_joint_attention.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_transformer/fibo_single_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_transformer/fibo_single_attention.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_transformer/joint_transformer_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_transformer/joint_transformer_block.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_transformer/single_transformer_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_transformer/single_transformer_block.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_transformer/text_projection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_transformer/text_projection.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_transformer/time_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_transformer/time_embed.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_transformer/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_transformer/transformer.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_vae/__init__.py: -------------------------------------------------------------------------------- 1 | """FIBO VAE components.""" 2 | -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_vae/common/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_vae/common/__init__.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_vae/common/wan_2_2_attention_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_vae/common/wan_2_2_attention_block.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_vae/common/wan_2_2_causal_conv_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_vae/common/wan_2_2_causal_conv_3d.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_vae/common/wan_2_2_mid_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_vae/common/wan_2_2_mid_block.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_vae/common/wan_2_2_resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_vae/common/wan_2_2_resample.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_vae/common/wan_2_2_residual_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_vae/common/wan_2_2_residual_block.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_vae/common/wan_2_2_rms_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_vae/common/wan_2_2_rms_norm.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_vae/decoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_vae/decoder/__init__.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_vae/decoder/wan_2_2_decoder_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_vae/decoder/wan_2_2_decoder_3d.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_vae/decoder/wan_2_2_dup_up_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_vae/decoder/wan_2_2_dup_up_3d.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_vae/decoder/wan_2_2_residual_up_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_vae/decoder/wan_2_2_residual_up_block.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_vae/decoder/wan_2_2_up_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_vae/decoder/wan_2_2_up_block.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_vae/encoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_vae/encoder/__init__.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_vae/encoder/wan_2_2_avg_down_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_vae/encoder/wan_2_2_avg_down_3d.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_vae/encoder/wan_2_2_down_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_vae/encoder/wan_2_2_down_block.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_vae/encoder/wan_2_2_encoder_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_vae/encoder/wan_2_2_encoder_3d.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/model/fibo_vae/wan_2_2_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/model/fibo_vae/wan_2_2_vae.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/tokenizer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/tokenizer/__init__.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/tokenizer/fibo_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/tokenizer/fibo_tokenizer.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/tokenizer/qwen2vl_image_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/tokenizer/qwen2vl_image_processor.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/tokenizer/qwen2vl_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/tokenizer/qwen2vl_processor.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/tokenizer/smol_lm3_3b_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/tokenizer/smol_lm3_3b_tokenizer.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/variants/__init__.py: -------------------------------------------------------------------------------- 1 | """FIBO model variants.""" 2 | -------------------------------------------------------------------------------- /src/mflux/models/fibo/variants/txt2img/__init__.py: -------------------------------------------------------------------------------- 1 | """FIBO text-to-image variant.""" 2 | -------------------------------------------------------------------------------- /src/mflux/models/fibo/variants/txt2img/fibo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/variants/txt2img/fibo.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/weights/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/weights/__init__.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/weights/fibo_weight_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/weights/fibo_weight_handler.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/weights/fibo_weight_mapping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/weights/fibo_weight_mapping.py -------------------------------------------------------------------------------- /src/mflux/models/fibo/weights/fibo_weight_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo/weights/fibo_weight_util.py -------------------------------------------------------------------------------- /src/mflux/models/fibo_vlm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/fibo_vlm/fibo_vlm_initializer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo_vlm/fibo_vlm_initializer.py -------------------------------------------------------------------------------- /src/mflux/models/fibo_vlm/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/fibo_vlm/model/fibo_vlm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo_vlm/model/fibo_vlm.py -------------------------------------------------------------------------------- /src/mflux/models/fibo_vlm/model/qwen3_vl_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo_vlm/model/qwen3_vl_attention.py -------------------------------------------------------------------------------- /src/mflux/models/fibo_vlm/model/qwen3_vl_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo_vlm/model/qwen3_vl_decoder.py -------------------------------------------------------------------------------- /src/mflux/models/fibo_vlm/model/qwen3_vl_decoder_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo_vlm/model/qwen3_vl_decoder_layer.py -------------------------------------------------------------------------------- /src/mflux/models/fibo_vlm/model/qwen3_vl_mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo_vlm/model/qwen3_vl_mlp.py -------------------------------------------------------------------------------- /src/mflux/models/fibo_vlm/model/qwen3_vl_rms_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo_vlm/model/qwen3_vl_rms_norm.py -------------------------------------------------------------------------------- /src/mflux/models/fibo_vlm/model/qwen3_vl_rope.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo_vlm/model/qwen3_vl_rope.py -------------------------------------------------------------------------------- /src/mflux/models/fibo_vlm/model/qwen3_vl_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo_vlm/model/qwen3_vl_util.py -------------------------------------------------------------------------------- /src/mflux/models/fibo_vlm/model/qwen3_vl_vision_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo_vlm/model/qwen3_vl_vision_attention.py -------------------------------------------------------------------------------- /src/mflux/models/fibo_vlm/model/qwen3_vl_vision_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo_vlm/model/qwen3_vl_vision_block.py -------------------------------------------------------------------------------- /src/mflux/models/fibo_vlm/model/qwen3_vl_vision_mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo_vlm/model/qwen3_vl_vision_mlp.py -------------------------------------------------------------------------------- /src/mflux/models/fibo_vlm/model/qwen3_vl_vision_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo_vlm/model/qwen3_vl_vision_model.py -------------------------------------------------------------------------------- /src/mflux/models/fibo_vlm/model/qwen3_vl_vision_patch_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo_vlm/model/qwen3_vl_vision_patch_embed.py -------------------------------------------------------------------------------- /src/mflux/models/fibo_vlm/model/qwen3_vl_vision_patch_merger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo_vlm/model/qwen3_vl_vision_patch_merger.py -------------------------------------------------------------------------------- /src/mflux/models/fibo_vlm/model/qwen3_vl_vision_rotary_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo_vlm/model/qwen3_vl_vision_rotary_embedding.py -------------------------------------------------------------------------------- /src/mflux/models/fibo_vlm/weights/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/fibo_vlm/weights/fibo_vlm_weight_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo_vlm/weights/fibo_vlm_weight_handler.py -------------------------------------------------------------------------------- /src/mflux/models/fibo_vlm/weights/fibo_vlm_weight_mapping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/fibo_vlm/weights/fibo_vlm_weight_mapping.py -------------------------------------------------------------------------------- /src/mflux/models/flux/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/flux_initializer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/flux_initializer.py -------------------------------------------------------------------------------- /src/mflux/models/flux/latent_creator/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/latent_creator/flux_latent_creator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/latent_creator/flux_latent_creator.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_text_encoder/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_text_encoder/clip_encoder/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_text_encoder/clip_encoder/clip_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_text_encoder/clip_encoder/clip_embeddings.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_text_encoder/clip_encoder/clip_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_text_encoder/clip_encoder/clip_encoder.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_text_encoder/clip_encoder/clip_encoder_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_text_encoder/clip_encoder/clip_encoder_layer.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_text_encoder/clip_encoder/clip_mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_text_encoder/clip_encoder/clip_mlp.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_text_encoder/clip_encoder/clip_sdpa_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_text_encoder/clip_encoder/clip_sdpa_attention.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_text_encoder/clip_encoder/clip_text_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_text_encoder/clip_encoder/clip_text_model.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_text_encoder/clip_encoder/encoder_clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_text_encoder/clip_encoder/encoder_clip.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_text_encoder/prompt_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_text_encoder/prompt_encoder.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_text_encoder/t5_encoder/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_text_encoder/t5_encoder/t5_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_text_encoder/t5_encoder/t5_attention.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_text_encoder/t5_encoder/t5_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_text_encoder/t5_encoder/t5_block.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_text_encoder/t5_encoder/t5_dense_relu_dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_text_encoder/t5_encoder/t5_dense_relu_dense.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_text_encoder/t5_encoder/t5_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_text_encoder/t5_encoder/t5_encoder.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_text_encoder/t5_encoder/t5_feed_forward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_text_encoder/t5_encoder/t5_feed_forward.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_text_encoder/t5_encoder/t5_layer_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_text_encoder/t5_encoder/t5_layer_norm.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_text_encoder/t5_encoder/t5_self_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_text_encoder/t5_encoder/t5_self_attention.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_transformer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_transformer/ada_layer_norm_continuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_transformer/ada_layer_norm_continuous.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_transformer/ada_layer_norm_zero.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_transformer/ada_layer_norm_zero.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_transformer/ada_layer_norm_zero_single.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_transformer/ada_layer_norm_zero_single.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_transformer/common/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_transformer/common/attention_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_transformer/common/attention_utils.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_transformer/embed_nd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_transformer/embed_nd.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_transformer/feed_forward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_transformer/feed_forward.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_transformer/guidance_embedder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_transformer/guidance_embedder.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_transformer/joint_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_transformer/joint_attention.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_transformer/joint_transformer_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_transformer/joint_transformer_block.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_transformer/single_block_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_transformer/single_block_attention.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_transformer/single_transformer_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_transformer/single_transformer_block.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_transformer/text_embedder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_transformer/text_embedder.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_transformer/time_text_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_transformer/time_text_embed.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_transformer/timestep_embedder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_transformer/timestep_embedder.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_transformer/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_transformer/transformer.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_vae/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_vae/common/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_vae/common/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_vae/common/attention.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_vae/common/resnet_block_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_vae/common/resnet_block_2d.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_vae/common/unet_mid_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_vae/common/unet_mid_block.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_vae/decoder/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_vae/decoder/conv_in.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_vae/decoder/conv_in.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_vae/decoder/conv_norm_out.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_vae/decoder/conv_norm_out.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_vae/decoder/conv_out.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_vae/decoder/conv_out.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_vae/decoder/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_vae/decoder/decoder.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_vae/decoder/up_block_1_or_2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_vae/decoder/up_block_1_or_2.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_vae/decoder/up_block_3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_vae/decoder/up_block_3.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_vae/decoder/up_block_4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_vae/decoder/up_block_4.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_vae/decoder/up_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_vae/decoder/up_sampler.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_vae/encoder/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_vae/encoder/conv_in.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_vae/encoder/conv_in.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_vae/encoder/conv_norm_out.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_vae/encoder/conv_norm_out.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_vae/encoder/conv_out.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_vae/encoder/conv_out.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_vae/encoder/down_block_1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_vae/encoder/down_block_1.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_vae/encoder/down_block_2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_vae/encoder/down_block_2.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_vae/encoder/down_block_3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_vae/encoder/down_block_3.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_vae/encoder/down_block_4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_vae/encoder/down_block_4.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_vae/encoder/down_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_vae/encoder/down_sampler.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_vae/encoder/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_vae/encoder/encoder.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/flux_vae/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/flux_vae/vae.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/redux_encoder/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/model/redux_encoder/redux_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/redux_encoder/redux_encoder.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/siglip_vision_transformer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/model/siglip_vision_transformer/siglip_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/siglip_vision_transformer/siglip_encoder.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/siglip_vision_transformer/siglip_encoder_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/siglip_vision_transformer/siglip_encoder_layer.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/siglip_vision_transformer/siglip_mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/siglip_vision_transformer/siglip_mlp.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/siglip_vision_transformer/siglip_multi_head_attention_pooling_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/siglip_vision_transformer/siglip_multi_head_attention_pooling_head.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/siglip_vision_transformer/siglip_sdpa_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/siglip_vision_transformer/siglip_sdpa_attention.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/siglip_vision_transformer/siglip_vision_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/siglip_vision_transformer/siglip_vision_embeddings.py -------------------------------------------------------------------------------- /src/mflux/models/flux/model/siglip_vision_transformer/siglip_vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/model/siglip_vision_transformer/siglip_vision_transformer.py -------------------------------------------------------------------------------- /src/mflux/models/flux/tokenizer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/tokenizer/clip_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/tokenizer/clip_tokenizer.py -------------------------------------------------------------------------------- /src/mflux/models/flux/tokenizer/t5_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/tokenizer/t5_tokenizer.py -------------------------------------------------------------------------------- /src/mflux/models/flux/tokenizer/tokenizer_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/tokenizer/tokenizer_handler.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/concept_attention/__init__.py: -------------------------------------------------------------------------------- 1 | # Concept Attention flux_transformer models 2 | -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/concept_attention/attention_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/concept_attention/attention_data.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/concept_attention/concept_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/concept_attention/concept_util.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/concept_attention/flux_concept.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/concept_attention/flux_concept.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/concept_attention/flux_concept_from_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/concept_attention/flux_concept_from_image.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/concept_attention/joint_attention_concept.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/concept_attention/joint_attention_concept.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/concept_attention/joint_transformer_block_concept.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/concept_attention/joint_transformer_block_concept.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/concept_attention/transformer_concept.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/concept_attention/transformer_concept.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/controlnet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/controlnet/controlnet_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/controlnet/controlnet_util.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/controlnet/flux_controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/controlnet/flux_controlnet.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/controlnet/transformer_controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/controlnet/transformer_controlnet.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/controlnet/weight_handler_controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/controlnet/weight_handler_controlnet.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/depth/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/depth/depth_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/depth/depth_util.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/depth/flux_depth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/depth/flux_depth.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/dreambooth/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/dreambooth/_example/images/01.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/dreambooth/_example/images/01.jpeg -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/dreambooth/_example/images/02.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/dreambooth/_example/images/02.jpeg -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/dreambooth/_example/images/03.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/dreambooth/_example/images/03.jpeg -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/dreambooth/_example/images/04.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/dreambooth/_example/images/04.jpeg -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/dreambooth/_example/images/05.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/dreambooth/_example/images/05.jpeg -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/dreambooth/_example/train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/dreambooth/_example/train.json -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/dreambooth/dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/dreambooth/dataset/batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/dreambooth/dataset/batch.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/dreambooth/dataset/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/dreambooth/dataset/dataset.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/dreambooth/dataset/dreambooth_preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/dreambooth/dataset/dreambooth_preprocessing.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/dreambooth/dataset/iterator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/dreambooth/dataset/iterator.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/dreambooth/dreambooth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/dreambooth/dreambooth.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/dreambooth/dreambooth_initializer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/dreambooth/dreambooth_initializer.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/dreambooth/lora_layers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/dreambooth/lora_layers/lora_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/dreambooth/lora_layers/lora_layers.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/dreambooth/optimization/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/dreambooth/optimization/_loss_derivation/dreambooth_loss.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/dreambooth/optimization/_loss_derivation/dreambooth_loss.pdf -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/dreambooth/optimization/_loss_derivation/dreambooth_loss.tex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/dreambooth/optimization/_loss_derivation/dreambooth_loss.tex -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/dreambooth/optimization/dreambooth_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/dreambooth/optimization/dreambooth_loss.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/dreambooth/optimization/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/dreambooth/optimization/optimizer.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/dreambooth/state/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/dreambooth/state/training_spec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/dreambooth/state/training_spec.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/dreambooth/state/training_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/dreambooth/state/training_state.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/dreambooth/state/zip_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/dreambooth/state/zip_util.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/dreambooth/statistics/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/dreambooth/statistics/plotter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/dreambooth/statistics/plotter.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/dreambooth/statistics/statistics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/dreambooth/statistics/statistics.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/fill/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/fill/flux_fill.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/fill/flux_fill.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/fill/mask_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/fill/mask_util.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/in_context/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/in_context/flux_in_context_dev.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/in_context/flux_in_context_dev.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/in_context/flux_in_context_fill.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/in_context/flux_in_context_fill.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/in_context/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/in_context/utils/in_context_loras.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/in_context/utils/in_context_loras.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/in_context/utils/in_context_mask_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/in_context/utils/in_context_mask_util.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/kontext/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/kontext/flux_kontext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/kontext/flux_kontext.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/kontext/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/kontext/utils/kontext_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/kontext/utils/kontext_util.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/redux/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/redux/flux_redux.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/redux/flux_redux.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/redux/redux_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/redux/redux_util.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/redux/weight_handler_redux.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/redux/weight_handler_redux.py -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/txt2img/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/variants/txt2img/flux.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/variants/txt2img/flux.py -------------------------------------------------------------------------------- /src/mflux/models/flux/weights/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/flux/weights/lora_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/weights/lora_converter.py -------------------------------------------------------------------------------- /src/mflux/models/flux/weights/weight_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/weights/weight_handler.py -------------------------------------------------------------------------------- /src/mflux/models/flux/weights/weight_handler_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/weights/weight_handler_lora.py -------------------------------------------------------------------------------- /src/mflux/models/flux/weights/weight_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/flux/weights/weight_util.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/qwen/latent_creator/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/qwen/latent_creator/qwen_latent_creator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/latent_creator/qwen_latent_creator.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_text_encoder/__init__.py: -------------------------------------------------------------------------------- 1 | # Qwen Text Encoder components 2 | -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_text_encoder/qwen_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_text_encoder/qwen_attention.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_text_encoder/qwen_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_text_encoder/qwen_encoder.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_text_encoder/qwen_encoder_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_text_encoder/qwen_encoder_layer.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_text_encoder/qwen_mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_text_encoder/qwen_mlp.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_text_encoder/qwen_patch_merger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_text_encoder/qwen_patch_merger.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_text_encoder/qwen_prompt_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_text_encoder/qwen_prompt_encoder.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_text_encoder/qwen_rms_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_text_encoder/qwen_rms_norm.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_text_encoder/qwen_rope.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_text_encoder/qwen_rope.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_text_encoder/qwen_text_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_text_encoder/qwen_text_encoder.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_text_encoder/qwen_vision_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_text_encoder/qwen_vision_attention.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_text_encoder/qwen_vision_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_text_encoder/qwen_vision_block.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_text_encoder/qwen_vision_language_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_text_encoder/qwen_vision_language_encoder.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_text_encoder/qwen_vision_language_prompt_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_text_encoder/qwen_vision_language_prompt_encoder.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_text_encoder/qwen_vision_mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_text_encoder/qwen_vision_mlp.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_text_encoder/qwen_vision_patch_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_text_encoder/qwen_vision_patch_embed.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_text_encoder/qwen_vision_rotary_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_text_encoder/qwen_vision_rotary_embedding.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_text_encoder/qwen_vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_text_encoder/qwen_vision_transformer.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_transformer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_transformer/qwen_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_transformer/qwen_attention.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_transformer/qwen_feed_forward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_transformer/qwen_feed_forward.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_transformer/qwen_layer_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_transformer/qwen_layer_norm.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_transformer/qwen_rope.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_transformer/qwen_rope.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_transformer/qwen_time_text_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_transformer/qwen_time_text_embed.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_transformer/qwen_timestep_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_transformer/qwen_timestep_embedding.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_transformer/qwen_timesteps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_transformer/qwen_timesteps.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_transformer/qwen_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_transformer/qwen_transformer.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_transformer/qwen_transformer_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_transformer/qwen_transformer_block.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_transformer/qwen_transformer_rms_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_transformer/qwen_transformer_rms_norm.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_vae/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_vae/qwen_image_attention_block_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_vae/qwen_image_attention_block_3d.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_vae/qwen_image_causal_conv_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_vae/qwen_image_causal_conv_3d.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_vae/qwen_image_decoder_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_vae/qwen_image_decoder_3d.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_vae/qwen_image_down_block_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_vae/qwen_image_down_block_3d.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_vae/qwen_image_encoder_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_vae/qwen_image_encoder_3d.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_vae/qwen_image_mid_block_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_vae/qwen_image_mid_block_3d.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_vae/qwen_image_res_block_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_vae/qwen_image_res_block_3d.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_vae/qwen_image_resample_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_vae/qwen_image_resample_3d.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_vae/qwen_image_rms_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_vae/qwen_image_rms_norm.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_vae/qwen_image_up_block_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_vae/qwen_image_up_block_3d.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/model/qwen_vae/qwen_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/model/qwen_vae/qwen_vae.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/qwen_edit_initializer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/qwen_edit_initializer.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/qwen_initializer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/qwen_initializer.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/tokenizer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/qwen/tokenizer/qwen_image_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/tokenizer/qwen_image_processor.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/tokenizer/qwen_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/tokenizer/qwen_tokenizer.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/tokenizer/qwen_tokenizer_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/tokenizer/qwen_tokenizer_handler.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/tokenizer/qwen_vision_language_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/tokenizer/qwen_vision_language_processor.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/tokenizer/qwen_vision_language_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/tokenizer/qwen_vision_language_tokenizer.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/variants/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/qwen/variants/edit/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/variants/edit/__init__.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/variants/edit/qwen_image_edit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/variants/edit/qwen_image_edit.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/variants/edit/utils/__init__.py: -------------------------------------------------------------------------------- 1 | # Qwen Edit utilities 2 | -------------------------------------------------------------------------------- /src/mflux/models/qwen/variants/edit/utils/qwen_edit_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/variants/edit/utils/qwen_edit_util.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/variants/edit/utils/qwen_prompt_rewriter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/variants/edit/utils/qwen_prompt_rewriter.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/variants/txt2img/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/qwen/variants/txt2img/qwen_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/variants/txt2img/qwen_image.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/weights/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/models/qwen/weights/qwen_lora_mapping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/weights/qwen_lora_mapping.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/weights/qwen_weight_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/weights/qwen_weight_handler.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/weights/qwen_weight_mapping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/weights/qwen_weight_mapping.py -------------------------------------------------------------------------------- /src/mflux/models/qwen/weights/qwen_weight_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/models/qwen/weights/qwen_weight_util.py -------------------------------------------------------------------------------- /src/mflux/refine_fibo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/refine_fibo.py -------------------------------------------------------------------------------- /src/mflux/release/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/release/changelog_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/release/changelog_parser.py -------------------------------------------------------------------------------- /src/mflux/release/git_operations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/release/git_operations.py -------------------------------------------------------------------------------- /src/mflux/release/github_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/release/github_api.py -------------------------------------------------------------------------------- /src/mflux/release/pypi_publisher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/release/pypi_publisher.py -------------------------------------------------------------------------------- /src/mflux/release/release_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/release/release_manager.py -------------------------------------------------------------------------------- /src/mflux/release/release_script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/release/release_script.py -------------------------------------------------------------------------------- /src/mflux/release/release_validator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/release/release_validator.py -------------------------------------------------------------------------------- /src/mflux/save.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/save.py -------------------------------------------------------------------------------- /src/mflux/save_depth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/save_depth.py -------------------------------------------------------------------------------- /src/mflux/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/train.py -------------------------------------------------------------------------------- /src/mflux/ui/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/ui/box_values.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/ui/box_values.py -------------------------------------------------------------------------------- /src/mflux/ui/cli/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/ui/cli/completions/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/ui/cli/completions/README.md -------------------------------------------------------------------------------- /src/mflux/ui/cli/completions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/ui/cli/completions/__init__.py -------------------------------------------------------------------------------- /src/mflux/ui/cli/completions/generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/ui/cli/completions/generator.py -------------------------------------------------------------------------------- /src/mflux/ui/cli/completions/install.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/ui/cli/completions/install.py -------------------------------------------------------------------------------- /src/mflux/ui/cli/parsers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/ui/cli/parsers.py -------------------------------------------------------------------------------- /src/mflux/ui/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/ui/defaults.py -------------------------------------------------------------------------------- /src/mflux/ui/prompt_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/ui/prompt_utils.py -------------------------------------------------------------------------------- /src/mflux/ui/scale_factor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/ui/scale_factor.py -------------------------------------------------------------------------------- /src/mflux/upscale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/upscale.py -------------------------------------------------------------------------------- /src/mflux/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mflux/utils/array_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/utils/array_util.py -------------------------------------------------------------------------------- /src/mflux/utils/download.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/utils/download.py -------------------------------------------------------------------------------- /src/mflux/utils/exceptions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/utils/exceptions.py -------------------------------------------------------------------------------- /src/mflux/utils/generated_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/utils/generated_image.py -------------------------------------------------------------------------------- /src/mflux/utils/image_compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/utils/image_compare.py -------------------------------------------------------------------------------- /src/mflux/utils/image_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/utils/image_util.py -------------------------------------------------------------------------------- /src/mflux/utils/metadata_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/utils/metadata_builder.py -------------------------------------------------------------------------------- /src/mflux/utils/metadata_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/utils/metadata_reader.py -------------------------------------------------------------------------------- /src/mflux/utils/version_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/src/mflux/utils/version_util.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/arg_parser/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/arg_parser/test_cli_argparser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/arg_parser/test_cli_argparser.py -------------------------------------------------------------------------------- /tests/arg_parser/test_lora_library_integration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/arg_parser/test_lora_library_integration.py -------------------------------------------------------------------------------- /tests/arg_parser/test_stdin_prompt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/arg_parser/test_stdin_prompt.py -------------------------------------------------------------------------------- /tests/arg_parser/test_upscale_argparser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/arg_parser/test_upscale_argparser.py -------------------------------------------------------------------------------- /tests/callbacks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/callbacks/test_battery_saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/callbacks/test_battery_saver.py -------------------------------------------------------------------------------- /tests/depth/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/depth/test_depth_pro.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/depth/test_depth_pro.py -------------------------------------------------------------------------------- /tests/dreambooth/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/dreambooth/config/train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/dreambooth/config/train.json -------------------------------------------------------------------------------- /tests/dreambooth/test_dreambooth_dataset_iterator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/dreambooth/test_dreambooth_dataset_iterator.py -------------------------------------------------------------------------------- /tests/dreambooth/test_resume_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/dreambooth/test_resume_training.py -------------------------------------------------------------------------------- /tests/dreambooth/test_train_and_load_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/dreambooth/test_train_and_load_weights.py -------------------------------------------------------------------------------- /tests/image_generation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/image_generation/helpers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/image_generation/helpers/image_generation_concept_test_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/image_generation/helpers/image_generation_concept_test_helper.py -------------------------------------------------------------------------------- /tests/image_generation/helpers/image_generation_controlnet_test_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/image_generation/helpers/image_generation_controlnet_test_helper.py -------------------------------------------------------------------------------- /tests/image_generation/helpers/image_generation_depth_test_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/image_generation/helpers/image_generation_depth_test_helper.py -------------------------------------------------------------------------------- /tests/image_generation/helpers/image_generation_edit_test_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/image_generation/helpers/image_generation_edit_test_helper.py -------------------------------------------------------------------------------- /tests/image_generation/helpers/image_generation_fibo_test_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/image_generation/helpers/image_generation_fibo_test_helper.py -------------------------------------------------------------------------------- /tests/image_generation/helpers/image_generation_fill_test_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/image_generation/helpers/image_generation_fill_test_helper.py -------------------------------------------------------------------------------- /tests/image_generation/helpers/image_generation_ic_edit_test_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/image_generation/helpers/image_generation_ic_edit_test_helper.py -------------------------------------------------------------------------------- /tests/image_generation/helpers/image_generation_in_context_test_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/image_generation/helpers/image_generation_in_context_test_helper.py -------------------------------------------------------------------------------- /tests/image_generation/helpers/image_generation_kontext_test_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/image_generation/helpers/image_generation_kontext_test_helper.py -------------------------------------------------------------------------------- /tests/image_generation/helpers/image_generation_redux_test_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/image_generation/helpers/image_generation_redux_test_helper.py -------------------------------------------------------------------------------- /tests/image_generation/helpers/image_generation_test_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/image_generation/helpers/image_generation_test_helper.py -------------------------------------------------------------------------------- /tests/image_generation/test_generate_concept.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/image_generation/test_generate_concept.py -------------------------------------------------------------------------------- /tests/image_generation/test_generate_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/image_generation/test_generate_image.py -------------------------------------------------------------------------------- /tests/image_generation/test_generate_image_controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/image_generation/test_generate_image_controlnet.py -------------------------------------------------------------------------------- /tests/image_generation/test_generate_image_depth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/image_generation/test_generate_image_depth.py -------------------------------------------------------------------------------- /tests/image_generation/test_generate_image_fibo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/image_generation/test_generate_image_fibo.py -------------------------------------------------------------------------------- /tests/image_generation/test_generate_image_fill.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/image_generation/test_generate_image_fill.py -------------------------------------------------------------------------------- /tests/image_generation/test_generate_image_in_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/image_generation/test_generate_image_in_context.py -------------------------------------------------------------------------------- /tests/image_generation/test_generate_image_kontext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/image_generation/test_generate_image_kontext.py -------------------------------------------------------------------------------- /tests/image_generation/test_generate_image_qwen_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/image_generation/test_generate_image_qwen_image.py -------------------------------------------------------------------------------- /tests/image_generation/test_generate_image_qwen_image_edit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/image_generation/test_generate_image_qwen_image_edit.py -------------------------------------------------------------------------------- /tests/image_generation/test_generate_image_redux.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/image_generation/test_generate_image_redux.py -------------------------------------------------------------------------------- /tests/image_generation/test_image_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/image_generation/test_image_util.py -------------------------------------------------------------------------------- /tests/image_generation/test_stdin_prompt_integration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/image_generation/test_stdin_prompt_integration.py -------------------------------------------------------------------------------- /tests/image_generation/test_upscale_dimensions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/image_generation/test_upscale_dimensions.py -------------------------------------------------------------------------------- /tests/metadata/__init__.py: -------------------------------------------------------------------------------- 1 | # Metadata tests 2 | 3 | -------------------------------------------------------------------------------- /tests/metadata/test_metadata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/metadata/test_metadata.py -------------------------------------------------------------------------------- /tests/model_config/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/model_config/test_model_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/model_config/test_model_config.py -------------------------------------------------------------------------------- /tests/model_saving/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/model_saving/test_model_saving.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/model_saving/test_model_saving.py -------------------------------------------------------------------------------- /tests/model_saving/test_model_saving_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/model_saving/test_model_saving_lora.py -------------------------------------------------------------------------------- /tests/resources/.gitignore: -------------------------------------------------------------------------------- 1 | output_*.png 2 | -------------------------------------------------------------------------------- /tests/resources/FLUX-dev-lora-MiaoKa-Yarn-World.safetensors: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/FLUX-dev-lora-MiaoKa-Yarn-World.safetensors -------------------------------------------------------------------------------- /tests/resources/Flux_-_Renaissance_art_style.safetensors: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/Flux_-_Renaissance_art_style.safetensors -------------------------------------------------------------------------------- /tests/resources/controlnet_reference.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/controlnet_reference.png -------------------------------------------------------------------------------- /tests/resources/ic_lora_reference_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/ic_lora_reference_logo.png -------------------------------------------------------------------------------- /tests/resources/low_res.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/low_res.jpg -------------------------------------------------------------------------------- /tests/resources/mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/mask.png -------------------------------------------------------------------------------- /tests/resources/reference_concept_from_image_schnell_heatmap.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/reference_concept_from_image_schnell_heatmap.png -------------------------------------------------------------------------------- /tests/resources/reference_concept_schnell_heatmap.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/reference_concept_schnell_heatmap.png -------------------------------------------------------------------------------- /tests/resources/reference_controlnet_dev.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/reference_controlnet_dev.png -------------------------------------------------------------------------------- /tests/resources/reference_controlnet_dev_lora.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/reference_controlnet_dev_lora.png -------------------------------------------------------------------------------- /tests/resources/reference_controlnet_dev_lora_depth.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/reference_controlnet_dev_lora_depth.png -------------------------------------------------------------------------------- /tests/resources/reference_controlnet_schnell.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/reference_controlnet_schnell.png -------------------------------------------------------------------------------- /tests/resources/reference_depth_dev_from_image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/reference_depth_dev_from_image.png -------------------------------------------------------------------------------- /tests/resources/reference_dev.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/reference_dev.png -------------------------------------------------------------------------------- /tests/resources/reference_dev_image_to_image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/reference_dev_image_to_image.png -------------------------------------------------------------------------------- /tests/resources/reference_dev_image_to_image_fill.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/reference_dev_image_to_image_fill.png -------------------------------------------------------------------------------- /tests/resources/reference_dev_kontext.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/reference_dev_kontext.png -------------------------------------------------------------------------------- /tests/resources/reference_dev_lora.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/reference_dev_lora.png -------------------------------------------------------------------------------- /tests/resources/reference_dev_lora_multiple.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/reference_dev_lora_multiple.png -------------------------------------------------------------------------------- /tests/resources/reference_fibo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/reference_fibo.png -------------------------------------------------------------------------------- /tests/resources/reference_fibo_white_owl.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/reference_fibo_white_owl.png -------------------------------------------------------------------------------- /tests/resources/reference_ic_edit.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/reference_ic_edit.png -------------------------------------------------------------------------------- /tests/resources/reference_ic_lora_identity.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/reference_ic_lora_identity.png -------------------------------------------------------------------------------- /tests/resources/reference_qwen_edit.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/reference_qwen_edit.png -------------------------------------------------------------------------------- /tests/resources/reference_qwen_edit_multiple_images.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/reference_qwen_edit_multiple_images.png -------------------------------------------------------------------------------- /tests/resources/reference_qwen_img2img.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/reference_qwen_img2img.png -------------------------------------------------------------------------------- /tests/resources/reference_qwen_lora.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/reference_qwen_lora.png -------------------------------------------------------------------------------- /tests/resources/reference_qwen_txt2img.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/reference_qwen_txt2img.png -------------------------------------------------------------------------------- /tests/resources/reference_redux_dev.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/reference_redux_dev.png -------------------------------------------------------------------------------- /tests/resources/reference_schnell.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/reference_schnell.png -------------------------------------------------------------------------------- /tests/resources/reference_upscaled.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/reference_upscaled.png -------------------------------------------------------------------------------- /tests/resources/shirt.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/shirt.jpg -------------------------------------------------------------------------------- /tests/resources/skyscrapers.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/resources/skyscrapers.jpg -------------------------------------------------------------------------------- /tests/schedulers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/schedulers/test_base_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/schedulers/test_base_scheduler.py -------------------------------------------------------------------------------- /tests/schedulers/test_linear_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/schedulers/test_linear_scheduler.py -------------------------------------------------------------------------------- /tests/schedulers/test_scheduler_lookup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/schedulers/test_scheduler_lookup.py -------------------------------------------------------------------------------- /tests/ui/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/ui/test_scale_factor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/ui/test_scale_factor.py -------------------------------------------------------------------------------- /tests/vlm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/vlm/test_fibo_vlm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/vlm/test_fibo_vlm.py -------------------------------------------------------------------------------- /tests/weights/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/weights/test_lora_library.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tests/weights/test_lora_library.py -------------------------------------------------------------------------------- /tools/create_outpaint_image_canvas_and_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tools/create_outpaint_image_canvas_and_mask.py -------------------------------------------------------------------------------- /tools/inpaint_mask_tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tools/inpaint_mask_tool.py -------------------------------------------------------------------------------- /tools/rename_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/tools/rename_images.py -------------------------------------------------------------------------------- /uv.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filipstrand/mflux/HEAD/uv.lock --------------------------------------------------------------------------------