├── .devcontainer └── devcontainer.json ├── .dockerignore ├── .github ├── ISSUE_TEMPLATE │ ├── bug_report.md │ ├── config.yml │ ├── feature_request.md │ └── general-questions.md ├── pull_request_template.md └── workflows │ ├── build.yml │ └── docs.yml ├── .gitignore ├── .pre-commit-config.yaml ├── .readthedocs.yaml ├── Dockerfile ├── LICENSE ├── MANIFEST.in ├── README.md ├── diffengine ├── __init__.py ├── configs │ ├── __init__.py │ ├── _base_ │ │ ├── datasets │ │ │ ├── cartoonization_xl.py │ │ │ ├── cat_waterpainting_dreambooth_pixart_1024.py │ │ │ ├── cat_waterpainting_dreambooth_xl.py │ │ │ ├── dog_dreambooth.py │ │ │ ├── dog_dreambooth_if.py │ │ │ ├── dog_dreambooth_pixart_1024.py │ │ │ ├── dog_dreambooth_pixart_512.py │ │ │ ├── dog_dreambooth_xl.py │ │ │ ├── dog_inpaint.py │ │ │ ├── dog_inpaint_multiple_mask.py │ │ │ ├── dog_inpaint_xl.py │ │ │ ├── face_spiga_controlnet.py │ │ │ ├── fill50k_controlnet.py │ │ │ ├── fill50k_controlnet_xl.py │ │ │ ├── fill50k_t2i_adapter_xl.py │ │ │ ├── gogh_esd_xl.py │ │ │ ├── google_backpack_dreambooth_xl.py │ │ │ ├── instructpix2pix_xl.py │ │ │ ├── keramer_face_dreambooth.py │ │ │ ├── keramer_face_dreambooth_xl.py │ │ │ ├── low_level_image_process_xl.py │ │ │ ├── pickascore_xl.py │ │ │ ├── pokemon_blip.py │ │ │ ├── pokemon_blip_amused_512.py │ │ │ ├── pokemon_blip_if.py │ │ │ ├── pokemon_blip_kandinsky_decoder.py │ │ │ ├── pokemon_blip_kandinsky_prior.py │ │ │ ├── pokemon_blip_kandinsky_v3.py │ │ │ ├── pokemon_blip_pixart.py │ │ │ ├── pokemon_blip_wuerstchen.py │ │ │ ├── pokemon_blip_xl.py │ │ │ ├── pokemon_blip_xl_ip_adapter.py │ │ │ ├── pokemon_blip_xl_ip_adapter_dinov2.py │ │ │ ├── pokemon_blip_xl_ip_adapter_dinov2_giant.py │ │ │ ├── pokemon_blip_xl_ip_adapter_eva02.py │ │ │ ├── pokemon_blip_xl_ip_adapter_siglip_384.py │ │ │ ├── pokemon_blip_xl_pre_compute.py │ │ │ ├── potatohead_dreambooth_xl.py │ │ │ ├── starbucks_dreambooth_xl.py │ │ │ └── waterpainting_xl.py │ │ ├── default_runtime.py │ │ ├── models │ │ │ ├── amused_512.py │ │ │ ├── deepfloyd_if_l.py │ │ │ ├── deepfloyd_if_xl_lora.py │ │ │ ├── distill_ssd_1b.py │ │ │ ├── distill_ssd_1b_from_sdxl.py │ │ │ ├── kandinsky_v22_decoder.py │ │ │ ├── kandinsky_v22_prior.py │ │ │ ├── kandinsky_v3.py │ │ │ ├── lcm_xl.py │ │ │ ├── lcm_xl_lora.py │ │ │ ├── pixart_alpha_1024.py │ │ │ ├── pixart_alpha_1024_lora.py │ │ │ ├── pixart_alpha_512_lora.py │ │ │ ├── small_sd_lora.py │ │ │ ├── small_sd_xl.py │ │ │ ├── ssd_1b_lora.py │ │ │ ├── stable_diffusion_inpaint.py │ │ │ ├── stable_diffusion_v15.py │ │ │ ├── stable_diffusion_v15_controlnet.py │ │ │ ├── stable_diffusion_v15_lora.py │ │ │ ├── stable_diffusion_v15_lora_textencoder.py │ │ │ ├── stable_diffusion_v15_textencoder.py │ │ │ ├── stable_diffusion_v21_lora.py │ │ │ ├── stable_diffusion_xl.py │ │ │ ├── stable_diffusion_xl_controlnet.py │ │ │ ├── stable_diffusion_xl_dpo.py │ │ │ ├── stable_diffusion_xl_esd.py │ │ │ ├── stable_diffusion_xl_inpaint.py │ │ │ ├── stable_diffusion_xl_instruct_pix2pix.py │ │ │ ├── stable_diffusion_xl_instruct_pix2pix_pretrained.py │ │ │ ├── stable_diffusion_xl_ip_adapter.py │ │ │ ├── stable_diffusion_xl_ip_adapter_plus.py │ │ │ ├── stable_diffusion_xl_ip_adapter_plus_eva02.py │ │ │ ├── stable_diffusion_xl_loha.py │ │ │ ├── stable_diffusion_xl_lokr.py │ │ │ ├── stable_diffusion_xl_lora.py │ │ │ ├── stable_diffusion_xl_oft.py │ │ │ ├── stable_diffusion_xl_t2i_adapter.py │ │ │ ├── tiny_sd_lora.py │ │ │ ├── tiny_sd_xl.py │ │ │ ├── wuerstchen_prior.py │ │ │ └── wuerstchen_prior_lora.py │ │ └── schedules │ │ │ ├── lcm_xl_50e.py │ │ │ ├── stable_diffusion_1e.py │ │ │ ├── stable_diffusion_1k.py │ │ │ ├── stable_diffusion_3e.py │ │ │ ├── stable_diffusion_500.py │ │ │ ├── stable_diffusion_50e.py │ │ │ └── stable_diffusion_xl_50e.py │ ├── amused │ │ ├── README.md │ │ └── amused_512_pokemon_blip.py │ ├── debias_estimation_loss │ │ ├── README.md │ │ └── stable_diffusion_xl_pokemon_blip_debias_estimation_loss.py │ ├── deepfloyd_if │ │ ├── README.md │ │ └── deepfloyd_if_l_pokemon_blip.py │ ├── deepfloyd_if_dreambooth │ │ ├── README.md │ │ └── deepfloyd_if_xl_dreambooth_lora_dog.py │ ├── diffusion_dpo │ │ ├── README.md │ │ └── stable_diffusion_xl_dpo_pickascore.py │ ├── distill_sd │ │ ├── README.md │ │ ├── small_sd_xl_pokemon_blip.py │ │ └── tiny_sd_xl_pokemon_blip.py │ ├── distill_sd_dreambooth │ │ ├── README.md │ │ ├── small_sd_dreambooth_lora_dog.py │ │ ├── small_sd_dreambooth_lora_keramer_face.py │ │ ├── tiny_sd_dreambooth_lora_dog.py │ │ └── tiny_sd_dreambooth_lora_keramer_face.py │ ├── esd │ │ ├── README.md │ │ └── stable_diffusion_xl_gogh_esd.py │ ├── input_perturbation │ │ ├── README.md │ │ └── stable_diffusion_xl_pokemon_blip_input_perturbation.py │ ├── instruct_pix2pix │ │ ├── README.md │ │ ├── stable_diffusion_xl_cartoonization.py │ │ ├── stable_diffusion_xl_instruct_pix2pix.py │ │ └── stable_diffusion_xl_low_level_image_process.py │ ├── ip_adapter │ │ ├── README.md │ │ ├── stable_diffusion_xl_pokemon_blip_ip_adapter.py │ │ ├── stable_diffusion_xl_pokemon_blip_ip_adapter_plus.py │ │ ├── stable_diffusion_xl_pokemon_blip_ip_adapter_plus_dinov2.py │ │ ├── stable_diffusion_xl_pokemon_blip_ip_adapter_plus_dinov2_giant.py │ │ ├── stable_diffusion_xl_pokemon_blip_ip_adapter_plus_dinov2_giant_lasthidden.py │ │ ├── stable_diffusion_xl_pokemon_blip_ip_adapter_plus_eva02.py │ │ ├── stable_diffusion_xl_pokemon_blip_ip_adapter_plus_pretrained.py │ │ └── stable_diffusion_xl_pokemon_blip_ip_adapter_plus_siglip.py │ ├── kandinsky_v22 │ │ ├── README.md │ │ ├── kandinsky_v22_decoder_pokemon_blip.py │ │ └── kandinsky_v22_prior_pokemon_blip.py │ ├── kandinsky_v3 │ │ ├── README.md │ │ └── kandinsky_v3_pokemon_blip.py │ ├── lcm │ │ ├── README.md │ │ └── lcm_xl_pokemon_blip.py │ ├── lcm_lora │ │ ├── README.md │ │ └── lcm_xl_lora_pokemon_blip.py │ ├── loha │ │ ├── README.md │ │ └── stable_diffusion_xl_loha_pokemon_blip.py │ ├── lokr │ │ ├── README.md │ │ └── stable_diffusion_xl_lokr_pokemon_blip.py │ ├── min_snr_loss │ │ ├── README.md │ │ └── stable_diffusion_v15_snr_pokemon_blip.py │ ├── offset_noise │ │ ├── README.md │ │ └── stable_diffusion_xl_pokemon_blip_offset_noise.py │ ├── oft │ │ ├── README.md │ │ └── stable_diffusion_xl_oft_pokemon_blip.py │ ├── pixart_alpha │ │ ├── README.md │ │ └── pixart_alpha_1024_pokemon_blip.py │ ├── pixart_alpha_dreambooth │ │ ├── README.md │ │ ├── pixart_alpha_1024_dreambooth_lora_cat_waterpainting.py │ │ ├── pixart_alpha_1024_dreambooth_lora_dog.py │ │ └── pixart_alpha_512_dreambooth_lora_dog.py │ ├── pixart_alpha_lora │ │ ├── README.md │ │ └── pixart_alpha_1024_lora_pokemon_blip.py │ ├── pyramid_noise │ │ ├── README.md │ │ └── stable_diffusion_xl_pokemon_blip_pyramid_noise.py │ ├── ssd_1b │ │ ├── README.md │ │ ├── ssd_1b_distill_from_sdxl_pokemon_blip.py │ │ └── ssd_1b_distill_pokemon_blip.py │ ├── ssd_1b_dreambooth │ │ ├── README.md │ │ └── ssd_1b_dreambooth_lora_dog.py │ ├── stable_diffusion │ │ ├── README.md │ │ ├── stable_diffusion_v15_ema_pokemon_blip.py │ │ ├── stable_diffusion_v15_pokemon_blip.py │ │ └── stable_diffusion_v15_textencoder_pokemon_blip.py │ ├── stable_diffusion_controlnet │ │ ├── README.md │ │ ├── stable_diffusion_v15_controlnet_face_spiga.py │ │ └── stable_diffusion_v15_controlnet_fill50k.py │ ├── stable_diffusion_dreambooth │ │ ├── README.md │ │ ├── stable_diffusion_v15_dreambooth_lora_dog.py │ │ └── stable_diffusion_v21_dreambooth_lora_dog.py │ ├── stable_diffusion_inpaint │ │ ├── README.md │ │ ├── stable_diffusion_inpaint_dog.py │ │ └── stable_diffusion_inpaint_dog_multi_mask.py │ ├── stable_diffusion_lora │ │ ├── README.md │ │ ├── stable_diffusion_v15_lora_pokemon_blip.py │ │ └── stable_diffusion_v15_lora_textencoder_pokemon_blip.py │ ├── stable_diffusion_xl │ │ ├── README.md │ │ ├── stable_diffusion_xl_pokemon_blip.py │ │ ├── stable_diffusion_xl_pokemon_blip_colossal.py │ │ ├── stable_diffusion_xl_pokemon_blip_deepspeed_stage2.py │ │ ├── stable_diffusion_xl_pokemon_blip_deepspeed_stage3.py │ │ ├── stable_diffusion_xl_pokemon_blip_fast.py │ │ ├── stable_diffusion_xl_pokemon_blip_pre_compute.py │ │ └── stable_diffusion_xl_pokemon_blip_xformers.py │ ├── stable_diffusion_xl_controlnet │ │ ├── README.md │ │ └── stable_diffusion_xl_controlnet_fill50k.py │ ├── stable_diffusion_xl_controlnet_small │ │ ├── README.md │ │ ├── stable_diffusion_xl_controlnet_mid_fill50k.py │ │ └── stable_diffusion_xl_controlnet_small_fill50k.py │ ├── stable_diffusion_xl_dreambooth │ │ ├── README.md │ │ ├── stable_diffusion_xl_dreambooth_lora_cat_waterpainting.py │ │ ├── stable_diffusion_xl_dreambooth_lora_dog.py │ │ ├── stable_diffusion_xl_dreambooth_lora_google_backpack.py │ │ ├── stable_diffusion_xl_dreambooth_lora_keramer_face.py │ │ ├── stable_diffusion_xl_dreambooth_lora_potatehead.py │ │ └── stable_diffusion_xl_dreambooth_lora_starbucks.py │ ├── stable_diffusion_xl_inpaint │ │ ├── README.md │ │ └── stable_diffusion_xl_inpaint_dog.py │ ├── stable_diffusion_xl_lora │ │ ├── README.md │ │ ├── stable_diffusion_xl_lora_conv1x1_pokemon_blip.py │ │ ├── stable_diffusion_xl_lora_conv3x3_pokemon_blip.py │ │ ├── stable_diffusion_xl_lora_pokemon_blip.py │ │ └── stable_diffusion_xl_lora_waterpainting.py │ ├── t2i_adapter │ │ ├── README.md │ │ └── stable_diffusion_xl_t2i_adapter_fill50k.py │ ├── timesteps_bias │ │ ├── README.md │ │ ├── stable_diffusion_xl_pokemon_blip_earlier_bias.py │ │ ├── stable_diffusion_xl_pokemon_blip_later_bias.py │ │ └── stable_diffusion_xl_pokemon_blip_range_bias.py │ ├── v_prediction │ │ ├── README.md │ │ └── stable_diffusion_xl_lora_v_prediction_pokemon_blip.py │ ├── wuerstchen │ │ ├── README.md │ │ └── wuerstchen_prior_pokemon_blip.py │ └── wuerstchen_lora │ │ ├── README.md │ │ └── wuerstchen_prior_lora_pokemon_blip.py ├── datasets │ ├── __init__.py │ ├── hf_controlnet_datasets.py │ ├── hf_datasets.py │ ├── hf_dpo_datasets.py │ ├── hf_dreambooth_datasets.py │ ├── hf_esd_datasets.py │ ├── samplers │ │ ├── __init__.py │ │ └── batch_sampler.py │ ├── transforms │ │ ├── __init__.py │ │ ├── base.py │ │ ├── dump_image.py │ │ ├── formatting.py │ │ ├── loading.py │ │ ├── processing.py │ │ └── wrappers.py │ └── utils.py ├── engine │ ├── __init__.py │ ├── hooks │ │ ├── __init__.py │ │ ├── compile_hook.py │ │ ├── controlnet_save_hook.py │ │ ├── fast_norm_hook.py │ │ ├── ip_adapter_save_hook.py │ │ ├── lcm_ema_update_hook.py │ │ ├── peft_save_hook.py │ │ ├── prior_save_hook.py │ │ ├── sd_checkpoint_hook.py │ │ ├── t2i_adapter_save_hook.py │ │ ├── transformer_checkpoint_hook.py │ │ ├── unet_ema_hook.py │ │ └── visualization_hook.py │ └── optimizers │ │ ├── __init__.py │ │ └── builder.py ├── entry_point.py ├── models │ ├── __init__.py │ ├── archs │ │ ├── __init__.py │ │ ├── ip_adapter.py │ │ └── peft.py │ ├── editors │ │ ├── __init__.py │ │ ├── amused │ │ │ ├── __init__.py │ │ │ ├── amused.py │ │ │ └── amused_data_preprocessor.py │ │ ├── deepfloyd_if │ │ │ ├── __init__.py │ │ │ └── deepfloyd_if.py │ │ ├── distill_sd │ │ │ ├── __init__.py │ │ │ └── distill_sd_xl.py │ │ ├── esd │ │ │ ├── __init__.py │ │ │ ├── esd_xl.py │ │ │ └── esd_xl_data_preprocessor.py │ │ ├── instruct_pix2pix │ │ │ ├── __init__.py │ │ │ └── instruct_pix2pix_xl.py │ │ ├── ip_adapter │ │ │ ├── __init__.py │ │ │ ├── ip_adapter_xl.py │ │ │ ├── ip_adapter_xl_data_preprocessor.py │ │ │ ├── ip_adapter_xl_timm.py │ │ │ └── pipeline.py │ │ ├── kandinsky │ │ │ ├── __init__.py │ │ │ ├── kandinskyv22_decoder.py │ │ │ ├── kandinskyv22_decoder_preprocessor.py │ │ │ ├── kandinskyv22_prior.py │ │ │ └── kandinskyv3.py │ │ ├── lcm │ │ │ ├── __init__.py │ │ │ ├── lcm_modules.py │ │ │ └── lcm_xl.py │ │ ├── pixart_alpha │ │ │ ├── __init__.py │ │ │ ├── pixart_alpha.py │ │ │ └── pixart_alpha_data_preprocessor.py │ │ ├── ssd_1b │ │ │ ├── __init__.py │ │ │ └── ssd_1b.py │ │ ├── stable_diffusion │ │ │ ├── __init__.py │ │ │ ├── sd_data_preprocessor.py │ │ │ └── stable_diffusion.py │ │ ├── stable_diffusion_controlnet │ │ │ ├── __init__.py │ │ │ ├── sd_controlnet_data_preprocessor.py │ │ │ └── stable_diffusion_controlnet.py │ │ ├── stable_diffusion_inpaint │ │ │ ├── __init__.py │ │ │ ├── sd_inpaint_data_preprocessor.py │ │ │ └── stable_diffusion_inpaint.py │ │ ├── stable_diffusion_xl │ │ │ ├── __init__.py │ │ │ ├── sdxl_data_preprocessor.py │ │ │ └── stable_diffusion_xl.py │ │ ├── stable_diffusion_xl_controlnet │ │ │ ├── __init__.py │ │ │ ├── sdxl_controlnet_data_preprocessor.py │ │ │ └── stable_diffusion_xl_controlnet.py │ │ ├── stable_diffusion_xl_dpo │ │ │ ├── __init__.py │ │ │ ├── sdxl_dpo_data_preprocessor.py │ │ │ └── stable_diffusion_xl_dpo.py │ │ ├── stable_diffusion_xl_inpaint │ │ │ ├── __init__.py │ │ │ ├── sdxl_inpaint_data_preprocessor.py │ │ │ └── stable_diffusion_xl_inpaint.py │ │ ├── t2i_adapter │ │ │ ├── __init__.py │ │ │ └── stable_diffusion_xl_t2i_adapter.py │ │ └── wuerstchen │ │ │ ├── __init__.py │ │ │ ├── efficient_net_encoder.py │ │ │ └── wuerstchen_prior.py │ ├── losses │ │ ├── __init__.py │ │ ├── base.py │ │ ├── cross_entropy_loss.py │ │ ├── debias_estimation_loss.py │ │ ├── hubar_loss.py │ │ ├── l2_loss.py │ │ ├── snr_l2_loss.py │ │ └── utils.py │ └── utils │ │ ├── __init__.py │ │ ├── noise.py │ │ └── timesteps.py ├── registry.py ├── tools │ ├── analysis_tools │ │ └── show_vae.py │ ├── copy_cfg.py │ ├── list_cfg.py │ ├── model_converters │ │ └── publish_model2diffusers.py │ ├── preprocess │ │ ├── bucket_ids.py │ │ └── csv_to_txt.py │ └── train.py └── version.py ├── docker-compose.yml ├── docs ├── Makefile └── source │ ├── _templates │ ├── 404.html │ └── python │ │ ├── attribute.rst │ │ ├── class.rst │ │ ├── data.rst │ │ ├── exception.rst │ │ ├── function.rst │ │ ├── method.rst │ │ ├── module.rst │ │ ├── package.rst │ │ └── property.rst │ ├── conf.py │ ├── get_started.md │ ├── index.rst │ ├── run_guides │ ├── inference.md │ ├── run_controlnet.md │ ├── run_controlnet_xl.md │ ├── run_dreambooth.md │ ├── run_dreambooth_xl.md │ ├── run_instruct_pix2pix.md │ ├── run_ip_adapter.md │ ├── run_kandinsky_v22.md │ ├── run_lcm.md │ ├── run_lcm_lora.md │ ├── run_lora.md │ ├── run_lora_xl.md │ ├── run_pixart_alpha.md │ ├── run_pixart_alpha_dreambooth.md │ ├── run_pixart_alpha_lora.md │ ├── run_sd.md │ ├── run_sdxl.md │ ├── run_t2i_adapter.md │ ├── run_wuerstchen.md │ └── run_wuerstchen_lora.md │ └── user_guides │ ├── aspect_ratio_bucketing.md │ ├── config.md │ └── dataset_prepare.md ├── projects ├── README.md ├── controlnetxs │ ├── README.md │ ├── _base_ │ │ └── stable_diffusion_xl_controlnetxs.py │ ├── modules │ │ ├── __init__.py │ │ ├── controlnetxs.py │ │ ├── pipeline_controlnet_xs_sd_xl.py │ │ └── stable_diffusion_xl_controlnetxs.py │ └── stable_diffusion_xl_controlnetxs_fill50k.py ├── face_expression │ ├── README.md │ ├── _base_ │ │ ├── anythingv5_lora_textencoder.py │ │ ├── counterfeit_xl_lora.py │ │ ├── face_expression_dataset.py │ │ └── face_expression_xl_dataset.py │ ├── stable_diffusion_v15_lora_face_expression.py │ └── stable_diffusion_xl_lora_face_expression.py └── tohoku_zunko_project │ ├── README.md │ ├── _base_ │ ├── anythingv5_lora.py │ ├── counterfeit_xl_lora.py │ ├── zunko_dreambooth.py │ └── zunko_dreambooth_xl.py │ ├── anythingv5_dreambooth_lora_zunko.py │ ├── counterfeit_xl_dreambooth_lora_zunko.py │ ├── stable_diffusion_v15_dreambooth_lora_zunko.py │ └── stable_diffusion_xl_dreambooth_lora_zunko.py ├── pyproject.toml └── tests ├── configs ├── ip_adapter.py ├── pixart.py ├── sd.py ├── sdcn.py ├── sdxl.py ├── sdxl_t2iadapter.py └── sdxlcn.py ├── test_datasets ├── test_hf_controlnet_datasets.py ├── test_hf_datasets.py ├── test_hf_dreambooth_datasets.py ├── test_hf_esd_datasets.py ├── test_samplers │ └── test_batch_sampler.py └── test_transforms │ ├── test_formatting.py │ ├── test_loading.py │ ├── test_processing.py │ └── test_wrappers.py ├── test_engine ├── test_hooks │ ├── test_compile_hook.py │ ├── test_controlnet_save_hook.py │ ├── test_fast_norm_hook.py │ ├── test_ip_adapter_save_hook.py │ ├── test_lcm_ema_update_hook.py │ ├── test_peft_save_hook.py │ ├── test_sd_checkpoint_hook.py │ ├── test_t2i_adapter_save_hook.py │ ├── test_transformer_checkpoint_hook.py │ ├── test_unet_ema_hook.py │ ├── test_visualization_hook.py │ └── test_wuerstchen_save_hook.py └── test_optimizers │ └── test_optimizer.py ├── test_models ├── test_archs.py ├── test_editors │ ├── test_amused │ │ └── test_amused.py │ ├── test_deepfloyd_if │ │ └── test_deepfloyd_if.py │ ├── test_distill_sd │ │ └── test_distill_sd.py │ ├── test_esd │ │ └── test_esd.py │ ├── test_instruct_pix2pix │ │ └── test_instruct_pix2pix.py │ ├── test_ip_adapter │ │ ├── test_ip_adapter.py │ │ ├── test_ip_adapter_plus.py │ │ └── test_ip_adapter_plus_timm.py │ ├── test_kandinsky │ │ ├── test_kandinsky_v3.py │ │ ├── test_kandinskyv22_decoder.py │ │ └── test_kandinskyv22_prior.py │ ├── test_lcm │ │ └── test_lcm_xl.py │ ├── test_pixart_alpha │ │ └── test_pixart_alpha.py │ ├── test_ssd_1b │ │ └── test_ssd_1b.py │ ├── test_stable_diffusion │ │ └── test_stable_diffusion.py │ ├── test_stable_diffusion_controlnet │ │ └── test_stable_diffusion_controlnet.py │ ├── test_stable_diffusion_inpaint │ │ └── test_stable_diffusion_inpaint.py │ ├── test_stable_diffusion_xl │ │ └── test_stable_diffusion_xl.py │ ├── test_stable_diffusion_xl_controlnet │ │ └── test_stable_diffusion_xl_controlnet.py │ ├── test_stable_diffusion_xl_dpo │ │ └── test_stable_diffusion_xl_dpo.py │ ├── test_stable_diffusion_xl_inpaint │ │ └── test_stable_diffusion_xl_inpaint.py │ ├── test_t2i_adapter │ │ └── test_stable_diffusion_xl_t2i_adapter.py │ └── test_wuerstchen │ │ └── test_wuerstchen_prior.py ├── test_losses.py └── test_utils │ ├── test_noise.py │ └── test_timesteps.py └── testdata ├── bucket_ids.pkl ├── color.jpg ├── cond.jpg ├── dataset ├── color.jpg ├── metadata.csv ├── metadata2.csv └── metadata_cn.csv └── dataset_db └── color.jpg /.devcontainer/devcontainer.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/.devcontainer/devcontainer.json -------------------------------------------------------------------------------- /.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/.dockerignore -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/bug_report.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/.github/ISSUE_TEMPLATE/bug_report.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/config.yml: -------------------------------------------------------------------------------- 1 | blank_issues_enabled: false 2 | -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature_request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/.github/ISSUE_TEMPLATE/feature_request.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/general-questions.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/.github/ISSUE_TEMPLATE/general-questions.md -------------------------------------------------------------------------------- /.github/pull_request_template.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/.github/pull_request_template.md -------------------------------------------------------------------------------- /.github/workflows/build.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/.github/workflows/build.yml -------------------------------------------------------------------------------- /.github/workflows/docs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/.github/workflows/docs.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.readthedocs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/.readthedocs.yaml -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/Dockerfile -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/README.md -------------------------------------------------------------------------------- /diffengine/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/__init__.py -------------------------------------------------------------------------------- /diffengine/configs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/__init__.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/cartoonization_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/cartoonization_xl.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/cat_waterpainting_dreambooth_pixart_1024.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/cat_waterpainting_dreambooth_pixart_1024.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/cat_waterpainting_dreambooth_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/cat_waterpainting_dreambooth_xl.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/dog_dreambooth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/dog_dreambooth.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/dog_dreambooth_if.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/dog_dreambooth_if.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/dog_dreambooth_pixart_1024.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/dog_dreambooth_pixart_1024.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/dog_dreambooth_pixart_512.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/dog_dreambooth_pixart_512.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/dog_dreambooth_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/dog_dreambooth_xl.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/dog_inpaint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/dog_inpaint.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/dog_inpaint_multiple_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/dog_inpaint_multiple_mask.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/dog_inpaint_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/dog_inpaint_xl.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/face_spiga_controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/face_spiga_controlnet.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/fill50k_controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/fill50k_controlnet.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/fill50k_controlnet_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/fill50k_controlnet_xl.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/fill50k_t2i_adapter_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/fill50k_t2i_adapter_xl.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/gogh_esd_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/gogh_esd_xl.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/google_backpack_dreambooth_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/google_backpack_dreambooth_xl.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/instructpix2pix_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/instructpix2pix_xl.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/keramer_face_dreambooth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/keramer_face_dreambooth.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/keramer_face_dreambooth_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/keramer_face_dreambooth_xl.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/low_level_image_process_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/low_level_image_process_xl.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/pickascore_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/pickascore_xl.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/pokemon_blip_amused_512.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/pokemon_blip_amused_512.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/pokemon_blip_if.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/pokemon_blip_if.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/pokemon_blip_kandinsky_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/pokemon_blip_kandinsky_decoder.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/pokemon_blip_kandinsky_prior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/pokemon_blip_kandinsky_prior.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/pokemon_blip_kandinsky_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/pokemon_blip_kandinsky_v3.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/pokemon_blip_pixart.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/pokemon_blip_pixart.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/pokemon_blip_wuerstchen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/pokemon_blip_wuerstchen.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/pokemon_blip_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/pokemon_blip_xl.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/pokemon_blip_xl_ip_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/pokemon_blip_xl_ip_adapter.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/pokemon_blip_xl_ip_adapter_dinov2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/pokemon_blip_xl_ip_adapter_dinov2.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/pokemon_blip_xl_ip_adapter_dinov2_giant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/pokemon_blip_xl_ip_adapter_dinov2_giant.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/pokemon_blip_xl_ip_adapter_eva02.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/pokemon_blip_xl_ip_adapter_eva02.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/pokemon_blip_xl_ip_adapter_siglip_384.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/pokemon_blip_xl_ip_adapter_siglip_384.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/pokemon_blip_xl_pre_compute.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/pokemon_blip_xl_pre_compute.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/potatohead_dreambooth_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/potatohead_dreambooth_xl.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/starbucks_dreambooth_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/starbucks_dreambooth_xl.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/datasets/waterpainting_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/datasets/waterpainting_xl.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/amused_512.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/amused_512.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/deepfloyd_if_l.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/deepfloyd_if_l.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/deepfloyd_if_xl_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/deepfloyd_if_xl_lora.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/distill_ssd_1b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/distill_ssd_1b.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/distill_ssd_1b_from_sdxl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/distill_ssd_1b_from_sdxl.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/kandinsky_v22_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/kandinsky_v22_decoder.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/kandinsky_v22_prior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/kandinsky_v22_prior.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/kandinsky_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/kandinsky_v3.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/lcm_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/lcm_xl.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/lcm_xl_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/lcm_xl_lora.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/pixart_alpha_1024.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/pixart_alpha_1024.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/pixart_alpha_1024_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/pixart_alpha_1024_lora.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/pixart_alpha_512_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/pixart_alpha_512_lora.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/small_sd_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/small_sd_lora.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/small_sd_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/small_sd_xl.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/ssd_1b_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/ssd_1b_lora.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/stable_diffusion_inpaint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/stable_diffusion_inpaint.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/stable_diffusion_v15.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/stable_diffusion_v15.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/stable_diffusion_v15_controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/stable_diffusion_v15_controlnet.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/stable_diffusion_v15_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/stable_diffusion_v15_lora.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/stable_diffusion_v15_lora_textencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/stable_diffusion_v15_lora_textencoder.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/stable_diffusion_v15_textencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/stable_diffusion_v15_textencoder.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/stable_diffusion_v21_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/stable_diffusion_v21_lora.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/stable_diffusion_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/stable_diffusion_xl.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/stable_diffusion_xl_controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/stable_diffusion_xl_controlnet.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/stable_diffusion_xl_dpo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/stable_diffusion_xl_dpo.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/stable_diffusion_xl_esd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/stable_diffusion_xl_esd.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/stable_diffusion_xl_inpaint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/stable_diffusion_xl_inpaint.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/stable_diffusion_xl_instruct_pix2pix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/stable_diffusion_xl_instruct_pix2pix.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/stable_diffusion_xl_instruct_pix2pix_pretrained.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/stable_diffusion_xl_instruct_pix2pix_pretrained.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/stable_diffusion_xl_ip_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/stable_diffusion_xl_ip_adapter.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/stable_diffusion_xl_ip_adapter_plus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/stable_diffusion_xl_ip_adapter_plus.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/stable_diffusion_xl_ip_adapter_plus_eva02.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/stable_diffusion_xl_ip_adapter_plus_eva02.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/stable_diffusion_xl_loha.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/stable_diffusion_xl_loha.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/stable_diffusion_xl_lokr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/stable_diffusion_xl_lokr.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/stable_diffusion_xl_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/stable_diffusion_xl_lora.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/stable_diffusion_xl_oft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/stable_diffusion_xl_oft.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/stable_diffusion_xl_t2i_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/stable_diffusion_xl_t2i_adapter.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/tiny_sd_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/tiny_sd_lora.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/tiny_sd_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/tiny_sd_xl.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/wuerstchen_prior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/wuerstchen_prior.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/models/wuerstchen_prior_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/models/wuerstchen_prior_lora.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/schedules/lcm_xl_50e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/schedules/lcm_xl_50e.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/schedules/stable_diffusion_1e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/schedules/stable_diffusion_1e.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/schedules/stable_diffusion_1k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/schedules/stable_diffusion_1k.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/schedules/stable_diffusion_3e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/schedules/stable_diffusion_3e.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/schedules/stable_diffusion_500.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/schedules/stable_diffusion_500.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/schedules/stable_diffusion_50e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/schedules/stable_diffusion_50e.py -------------------------------------------------------------------------------- /diffengine/configs/_base_/schedules/stable_diffusion_xl_50e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/_base_/schedules/stable_diffusion_xl_50e.py -------------------------------------------------------------------------------- /diffengine/configs/amused/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/amused/README.md -------------------------------------------------------------------------------- /diffengine/configs/amused/amused_512_pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/amused/amused_512_pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/configs/debias_estimation_loss/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/debias_estimation_loss/README.md -------------------------------------------------------------------------------- /diffengine/configs/debias_estimation_loss/stable_diffusion_xl_pokemon_blip_debias_estimation_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/debias_estimation_loss/stable_diffusion_xl_pokemon_blip_debias_estimation_loss.py -------------------------------------------------------------------------------- /diffengine/configs/deepfloyd_if/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/deepfloyd_if/README.md -------------------------------------------------------------------------------- /diffengine/configs/deepfloyd_if/deepfloyd_if_l_pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/deepfloyd_if/deepfloyd_if_l_pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/configs/deepfloyd_if_dreambooth/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/deepfloyd_if_dreambooth/README.md -------------------------------------------------------------------------------- /diffengine/configs/deepfloyd_if_dreambooth/deepfloyd_if_xl_dreambooth_lora_dog.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/deepfloyd_if_dreambooth/deepfloyd_if_xl_dreambooth_lora_dog.py -------------------------------------------------------------------------------- /diffengine/configs/diffusion_dpo/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/diffusion_dpo/README.md -------------------------------------------------------------------------------- /diffengine/configs/diffusion_dpo/stable_diffusion_xl_dpo_pickascore.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/diffusion_dpo/stable_diffusion_xl_dpo_pickascore.py -------------------------------------------------------------------------------- /diffengine/configs/distill_sd/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/distill_sd/README.md -------------------------------------------------------------------------------- /diffengine/configs/distill_sd/small_sd_xl_pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/distill_sd/small_sd_xl_pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/configs/distill_sd/tiny_sd_xl_pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/distill_sd/tiny_sd_xl_pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/configs/distill_sd_dreambooth/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/distill_sd_dreambooth/README.md -------------------------------------------------------------------------------- /diffengine/configs/distill_sd_dreambooth/small_sd_dreambooth_lora_dog.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/distill_sd_dreambooth/small_sd_dreambooth_lora_dog.py -------------------------------------------------------------------------------- /diffengine/configs/distill_sd_dreambooth/small_sd_dreambooth_lora_keramer_face.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/distill_sd_dreambooth/small_sd_dreambooth_lora_keramer_face.py -------------------------------------------------------------------------------- /diffengine/configs/distill_sd_dreambooth/tiny_sd_dreambooth_lora_dog.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/distill_sd_dreambooth/tiny_sd_dreambooth_lora_dog.py -------------------------------------------------------------------------------- /diffengine/configs/distill_sd_dreambooth/tiny_sd_dreambooth_lora_keramer_face.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/distill_sd_dreambooth/tiny_sd_dreambooth_lora_keramer_face.py -------------------------------------------------------------------------------- /diffengine/configs/esd/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/esd/README.md -------------------------------------------------------------------------------- /diffengine/configs/esd/stable_diffusion_xl_gogh_esd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/esd/stable_diffusion_xl_gogh_esd.py -------------------------------------------------------------------------------- /diffengine/configs/input_perturbation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/input_perturbation/README.md -------------------------------------------------------------------------------- /diffengine/configs/input_perturbation/stable_diffusion_xl_pokemon_blip_input_perturbation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/input_perturbation/stable_diffusion_xl_pokemon_blip_input_perturbation.py -------------------------------------------------------------------------------- /diffengine/configs/instruct_pix2pix/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/instruct_pix2pix/README.md -------------------------------------------------------------------------------- /diffengine/configs/instruct_pix2pix/stable_diffusion_xl_cartoonization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/instruct_pix2pix/stable_diffusion_xl_cartoonization.py -------------------------------------------------------------------------------- /diffengine/configs/instruct_pix2pix/stable_diffusion_xl_instruct_pix2pix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/instruct_pix2pix/stable_diffusion_xl_instruct_pix2pix.py -------------------------------------------------------------------------------- /diffengine/configs/instruct_pix2pix/stable_diffusion_xl_low_level_image_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/instruct_pix2pix/stable_diffusion_xl_low_level_image_process.py -------------------------------------------------------------------------------- /diffengine/configs/ip_adapter/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/ip_adapter/README.md -------------------------------------------------------------------------------- /diffengine/configs/ip_adapter/stable_diffusion_xl_pokemon_blip_ip_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/ip_adapter/stable_diffusion_xl_pokemon_blip_ip_adapter.py -------------------------------------------------------------------------------- /diffengine/configs/ip_adapter/stable_diffusion_xl_pokemon_blip_ip_adapter_plus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/ip_adapter/stable_diffusion_xl_pokemon_blip_ip_adapter_plus.py -------------------------------------------------------------------------------- /diffengine/configs/ip_adapter/stable_diffusion_xl_pokemon_blip_ip_adapter_plus_dinov2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/ip_adapter/stable_diffusion_xl_pokemon_blip_ip_adapter_plus_dinov2.py -------------------------------------------------------------------------------- /diffengine/configs/ip_adapter/stable_diffusion_xl_pokemon_blip_ip_adapter_plus_dinov2_giant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/ip_adapter/stable_diffusion_xl_pokemon_blip_ip_adapter_plus_dinov2_giant.py -------------------------------------------------------------------------------- /diffengine/configs/ip_adapter/stable_diffusion_xl_pokemon_blip_ip_adapter_plus_dinov2_giant_lasthidden.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/ip_adapter/stable_diffusion_xl_pokemon_blip_ip_adapter_plus_dinov2_giant_lasthidden.py -------------------------------------------------------------------------------- /diffengine/configs/ip_adapter/stable_diffusion_xl_pokemon_blip_ip_adapter_plus_eva02.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/ip_adapter/stable_diffusion_xl_pokemon_blip_ip_adapter_plus_eva02.py -------------------------------------------------------------------------------- /diffengine/configs/ip_adapter/stable_diffusion_xl_pokemon_blip_ip_adapter_plus_pretrained.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/ip_adapter/stable_diffusion_xl_pokemon_blip_ip_adapter_plus_pretrained.py -------------------------------------------------------------------------------- /diffengine/configs/ip_adapter/stable_diffusion_xl_pokemon_blip_ip_adapter_plus_siglip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/ip_adapter/stable_diffusion_xl_pokemon_blip_ip_adapter_plus_siglip.py -------------------------------------------------------------------------------- /diffengine/configs/kandinsky_v22/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/kandinsky_v22/README.md -------------------------------------------------------------------------------- /diffengine/configs/kandinsky_v22/kandinsky_v22_decoder_pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/kandinsky_v22/kandinsky_v22_decoder_pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/configs/kandinsky_v22/kandinsky_v22_prior_pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/kandinsky_v22/kandinsky_v22_prior_pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/configs/kandinsky_v3/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/kandinsky_v3/README.md -------------------------------------------------------------------------------- /diffengine/configs/kandinsky_v3/kandinsky_v3_pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/kandinsky_v3/kandinsky_v3_pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/configs/lcm/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/lcm/README.md -------------------------------------------------------------------------------- /diffengine/configs/lcm/lcm_xl_pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/lcm/lcm_xl_pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/configs/lcm_lora/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/lcm_lora/README.md -------------------------------------------------------------------------------- /diffengine/configs/lcm_lora/lcm_xl_lora_pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/lcm_lora/lcm_xl_lora_pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/configs/loha/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/loha/README.md -------------------------------------------------------------------------------- /diffengine/configs/loha/stable_diffusion_xl_loha_pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/loha/stable_diffusion_xl_loha_pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/configs/lokr/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/lokr/README.md -------------------------------------------------------------------------------- /diffengine/configs/lokr/stable_diffusion_xl_lokr_pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/lokr/stable_diffusion_xl_lokr_pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/configs/min_snr_loss/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/min_snr_loss/README.md -------------------------------------------------------------------------------- /diffengine/configs/min_snr_loss/stable_diffusion_v15_snr_pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/min_snr_loss/stable_diffusion_v15_snr_pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/configs/offset_noise/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/offset_noise/README.md -------------------------------------------------------------------------------- /diffengine/configs/offset_noise/stable_diffusion_xl_pokemon_blip_offset_noise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/offset_noise/stable_diffusion_xl_pokemon_blip_offset_noise.py -------------------------------------------------------------------------------- /diffengine/configs/oft/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/oft/README.md -------------------------------------------------------------------------------- /diffengine/configs/oft/stable_diffusion_xl_oft_pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/oft/stable_diffusion_xl_oft_pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/configs/pixart_alpha/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/pixart_alpha/README.md -------------------------------------------------------------------------------- /diffengine/configs/pixart_alpha/pixart_alpha_1024_pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/pixart_alpha/pixart_alpha_1024_pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/configs/pixart_alpha_dreambooth/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/pixart_alpha_dreambooth/README.md -------------------------------------------------------------------------------- /diffengine/configs/pixart_alpha_dreambooth/pixart_alpha_1024_dreambooth_lora_cat_waterpainting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/pixart_alpha_dreambooth/pixart_alpha_1024_dreambooth_lora_cat_waterpainting.py -------------------------------------------------------------------------------- /diffengine/configs/pixart_alpha_dreambooth/pixart_alpha_1024_dreambooth_lora_dog.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/pixart_alpha_dreambooth/pixart_alpha_1024_dreambooth_lora_dog.py -------------------------------------------------------------------------------- /diffengine/configs/pixart_alpha_dreambooth/pixart_alpha_512_dreambooth_lora_dog.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/pixart_alpha_dreambooth/pixart_alpha_512_dreambooth_lora_dog.py -------------------------------------------------------------------------------- /diffengine/configs/pixart_alpha_lora/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/pixart_alpha_lora/README.md -------------------------------------------------------------------------------- /diffengine/configs/pixart_alpha_lora/pixart_alpha_1024_lora_pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/pixart_alpha_lora/pixart_alpha_1024_lora_pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/configs/pyramid_noise/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/pyramid_noise/README.md -------------------------------------------------------------------------------- /diffengine/configs/pyramid_noise/stable_diffusion_xl_pokemon_blip_pyramid_noise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/pyramid_noise/stable_diffusion_xl_pokemon_blip_pyramid_noise.py -------------------------------------------------------------------------------- /diffengine/configs/ssd_1b/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/ssd_1b/README.md -------------------------------------------------------------------------------- /diffengine/configs/ssd_1b/ssd_1b_distill_from_sdxl_pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/ssd_1b/ssd_1b_distill_from_sdxl_pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/configs/ssd_1b/ssd_1b_distill_pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/ssd_1b/ssd_1b_distill_pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/configs/ssd_1b_dreambooth/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/ssd_1b_dreambooth/README.md -------------------------------------------------------------------------------- /diffengine/configs/ssd_1b_dreambooth/ssd_1b_dreambooth_lora_dog.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/ssd_1b_dreambooth/ssd_1b_dreambooth_lora_dog.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion/README.md -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion/stable_diffusion_v15_ema_pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion/stable_diffusion_v15_ema_pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion/stable_diffusion_v15_pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion/stable_diffusion_v15_pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion/stable_diffusion_v15_textencoder_pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion/stable_diffusion_v15_textencoder_pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_controlnet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_controlnet/README.md -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_controlnet/stable_diffusion_v15_controlnet_face_spiga.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_controlnet/stable_diffusion_v15_controlnet_face_spiga.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_controlnet/stable_diffusion_v15_controlnet_fill50k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_controlnet/stable_diffusion_v15_controlnet_fill50k.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_dreambooth/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_dreambooth/README.md -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_dreambooth/stable_diffusion_v15_dreambooth_lora_dog.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_dreambooth/stable_diffusion_v15_dreambooth_lora_dog.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_dreambooth/stable_diffusion_v21_dreambooth_lora_dog.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_dreambooth/stable_diffusion_v21_dreambooth_lora_dog.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_inpaint/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_inpaint/README.md -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_inpaint/stable_diffusion_inpaint_dog.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_inpaint/stable_diffusion_inpaint_dog.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_inpaint/stable_diffusion_inpaint_dog_multi_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_inpaint/stable_diffusion_inpaint_dog_multi_mask.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_lora/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_lora/README.md -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_lora/stable_diffusion_v15_lora_pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_lora/stable_diffusion_v15_lora_pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_lora/stable_diffusion_v15_lora_textencoder_pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_lora/stable_diffusion_v15_lora_textencoder_pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_xl/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_xl/README.md -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_xl/stable_diffusion_xl_pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_xl/stable_diffusion_xl_pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_xl/stable_diffusion_xl_pokemon_blip_colossal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_xl/stable_diffusion_xl_pokemon_blip_colossal.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_xl/stable_diffusion_xl_pokemon_blip_deepspeed_stage2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_xl/stable_diffusion_xl_pokemon_blip_deepspeed_stage2.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_xl/stable_diffusion_xl_pokemon_blip_deepspeed_stage3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_xl/stable_diffusion_xl_pokemon_blip_deepspeed_stage3.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_xl/stable_diffusion_xl_pokemon_blip_fast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_xl/stable_diffusion_xl_pokemon_blip_fast.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_xl/stable_diffusion_xl_pokemon_blip_pre_compute.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_xl/stable_diffusion_xl_pokemon_blip_pre_compute.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_xl/stable_diffusion_xl_pokemon_blip_xformers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_xl/stable_diffusion_xl_pokemon_blip_xformers.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_xl_controlnet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_xl_controlnet/README.md -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_xl_controlnet/stable_diffusion_xl_controlnet_fill50k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_xl_controlnet/stable_diffusion_xl_controlnet_fill50k.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_xl_controlnet_small/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_xl_controlnet_small/README.md -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_xl_controlnet_small/stable_diffusion_xl_controlnet_mid_fill50k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_xl_controlnet_small/stable_diffusion_xl_controlnet_mid_fill50k.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_xl_controlnet_small/stable_diffusion_xl_controlnet_small_fill50k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_xl_controlnet_small/stable_diffusion_xl_controlnet_small_fill50k.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_xl_dreambooth/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_xl_dreambooth/README.md -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_xl_dreambooth/stable_diffusion_xl_dreambooth_lora_cat_waterpainting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_xl_dreambooth/stable_diffusion_xl_dreambooth_lora_cat_waterpainting.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_xl_dreambooth/stable_diffusion_xl_dreambooth_lora_dog.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_xl_dreambooth/stable_diffusion_xl_dreambooth_lora_dog.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_xl_dreambooth/stable_diffusion_xl_dreambooth_lora_google_backpack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_xl_dreambooth/stable_diffusion_xl_dreambooth_lora_google_backpack.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_xl_dreambooth/stable_diffusion_xl_dreambooth_lora_keramer_face.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_xl_dreambooth/stable_diffusion_xl_dreambooth_lora_keramer_face.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_xl_dreambooth/stable_diffusion_xl_dreambooth_lora_potatehead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_xl_dreambooth/stable_diffusion_xl_dreambooth_lora_potatehead.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_xl_dreambooth/stable_diffusion_xl_dreambooth_lora_starbucks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_xl_dreambooth/stable_diffusion_xl_dreambooth_lora_starbucks.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_xl_inpaint/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_xl_inpaint/README.md -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_xl_inpaint/stable_diffusion_xl_inpaint_dog.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_xl_inpaint/stable_diffusion_xl_inpaint_dog.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_xl_lora/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_xl_lora/README.md -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_xl_lora/stable_diffusion_xl_lora_conv1x1_pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_xl_lora/stable_diffusion_xl_lora_conv1x1_pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_xl_lora/stable_diffusion_xl_lora_conv3x3_pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_xl_lora/stable_diffusion_xl_lora_conv3x3_pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_xl_lora/stable_diffusion_xl_lora_pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_xl_lora/stable_diffusion_xl_lora_pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/configs/stable_diffusion_xl_lora/stable_diffusion_xl_lora_waterpainting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/stable_diffusion_xl_lora/stable_diffusion_xl_lora_waterpainting.py -------------------------------------------------------------------------------- /diffengine/configs/t2i_adapter/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/t2i_adapter/README.md -------------------------------------------------------------------------------- /diffengine/configs/t2i_adapter/stable_diffusion_xl_t2i_adapter_fill50k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/t2i_adapter/stable_diffusion_xl_t2i_adapter_fill50k.py -------------------------------------------------------------------------------- /diffengine/configs/timesteps_bias/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/timesteps_bias/README.md -------------------------------------------------------------------------------- /diffengine/configs/timesteps_bias/stable_diffusion_xl_pokemon_blip_earlier_bias.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/timesteps_bias/stable_diffusion_xl_pokemon_blip_earlier_bias.py -------------------------------------------------------------------------------- /diffengine/configs/timesteps_bias/stable_diffusion_xl_pokemon_blip_later_bias.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/timesteps_bias/stable_diffusion_xl_pokemon_blip_later_bias.py -------------------------------------------------------------------------------- /diffengine/configs/timesteps_bias/stable_diffusion_xl_pokemon_blip_range_bias.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/timesteps_bias/stable_diffusion_xl_pokemon_blip_range_bias.py -------------------------------------------------------------------------------- /diffengine/configs/v_prediction/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/v_prediction/README.md -------------------------------------------------------------------------------- /diffengine/configs/v_prediction/stable_diffusion_xl_lora_v_prediction_pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/v_prediction/stable_diffusion_xl_lora_v_prediction_pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/configs/wuerstchen/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/wuerstchen/README.md -------------------------------------------------------------------------------- /diffengine/configs/wuerstchen/wuerstchen_prior_pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/wuerstchen/wuerstchen_prior_pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/configs/wuerstchen_lora/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/wuerstchen_lora/README.md -------------------------------------------------------------------------------- /diffengine/configs/wuerstchen_lora/wuerstchen_prior_lora_pokemon_blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/configs/wuerstchen_lora/wuerstchen_prior_lora_pokemon_blip.py -------------------------------------------------------------------------------- /diffengine/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/datasets/__init__.py -------------------------------------------------------------------------------- /diffengine/datasets/hf_controlnet_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/datasets/hf_controlnet_datasets.py -------------------------------------------------------------------------------- /diffengine/datasets/hf_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/datasets/hf_datasets.py -------------------------------------------------------------------------------- /diffengine/datasets/hf_dpo_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/datasets/hf_dpo_datasets.py -------------------------------------------------------------------------------- /diffengine/datasets/hf_dreambooth_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/datasets/hf_dreambooth_datasets.py -------------------------------------------------------------------------------- /diffengine/datasets/hf_esd_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/datasets/hf_esd_datasets.py -------------------------------------------------------------------------------- /diffengine/datasets/samplers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/datasets/samplers/__init__.py -------------------------------------------------------------------------------- /diffengine/datasets/samplers/batch_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/datasets/samplers/batch_sampler.py -------------------------------------------------------------------------------- /diffengine/datasets/transforms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/datasets/transforms/__init__.py -------------------------------------------------------------------------------- /diffengine/datasets/transforms/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/datasets/transforms/base.py -------------------------------------------------------------------------------- /diffengine/datasets/transforms/dump_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/datasets/transforms/dump_image.py -------------------------------------------------------------------------------- /diffengine/datasets/transforms/formatting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/datasets/transforms/formatting.py -------------------------------------------------------------------------------- /diffengine/datasets/transforms/loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/datasets/transforms/loading.py -------------------------------------------------------------------------------- /diffengine/datasets/transforms/processing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/datasets/transforms/processing.py -------------------------------------------------------------------------------- /diffengine/datasets/transforms/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/datasets/transforms/wrappers.py -------------------------------------------------------------------------------- /diffengine/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/datasets/utils.py -------------------------------------------------------------------------------- /diffengine/engine/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/engine/__init__.py -------------------------------------------------------------------------------- /diffengine/engine/hooks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/engine/hooks/__init__.py -------------------------------------------------------------------------------- /diffengine/engine/hooks/compile_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/engine/hooks/compile_hook.py -------------------------------------------------------------------------------- /diffengine/engine/hooks/controlnet_save_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/engine/hooks/controlnet_save_hook.py -------------------------------------------------------------------------------- /diffengine/engine/hooks/fast_norm_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/engine/hooks/fast_norm_hook.py -------------------------------------------------------------------------------- /diffengine/engine/hooks/ip_adapter_save_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/engine/hooks/ip_adapter_save_hook.py -------------------------------------------------------------------------------- /diffengine/engine/hooks/lcm_ema_update_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/engine/hooks/lcm_ema_update_hook.py -------------------------------------------------------------------------------- /diffengine/engine/hooks/peft_save_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/engine/hooks/peft_save_hook.py -------------------------------------------------------------------------------- /diffengine/engine/hooks/prior_save_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/engine/hooks/prior_save_hook.py -------------------------------------------------------------------------------- /diffengine/engine/hooks/sd_checkpoint_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/engine/hooks/sd_checkpoint_hook.py -------------------------------------------------------------------------------- /diffengine/engine/hooks/t2i_adapter_save_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/engine/hooks/t2i_adapter_save_hook.py -------------------------------------------------------------------------------- /diffengine/engine/hooks/transformer_checkpoint_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/engine/hooks/transformer_checkpoint_hook.py -------------------------------------------------------------------------------- /diffengine/engine/hooks/unet_ema_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/engine/hooks/unet_ema_hook.py -------------------------------------------------------------------------------- /diffengine/engine/hooks/visualization_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/engine/hooks/visualization_hook.py -------------------------------------------------------------------------------- /diffengine/engine/optimizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/engine/optimizers/__init__.py -------------------------------------------------------------------------------- /diffengine/engine/optimizers/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/engine/optimizers/builder.py -------------------------------------------------------------------------------- /diffengine/entry_point.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/entry_point.py -------------------------------------------------------------------------------- /diffengine/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/__init__.py -------------------------------------------------------------------------------- /diffengine/models/archs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/archs/__init__.py -------------------------------------------------------------------------------- /diffengine/models/archs/ip_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/archs/ip_adapter.py -------------------------------------------------------------------------------- /diffengine/models/archs/peft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/archs/peft.py -------------------------------------------------------------------------------- /diffengine/models/editors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/__init__.py -------------------------------------------------------------------------------- /diffengine/models/editors/amused/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/amused/__init__.py -------------------------------------------------------------------------------- /diffengine/models/editors/amused/amused.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/amused/amused.py -------------------------------------------------------------------------------- /diffengine/models/editors/amused/amused_data_preprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/amused/amused_data_preprocessor.py -------------------------------------------------------------------------------- /diffengine/models/editors/deepfloyd_if/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/deepfloyd_if/__init__.py -------------------------------------------------------------------------------- /diffengine/models/editors/deepfloyd_if/deepfloyd_if.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/deepfloyd_if/deepfloyd_if.py -------------------------------------------------------------------------------- /diffengine/models/editors/distill_sd/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/distill_sd/__init__.py -------------------------------------------------------------------------------- /diffengine/models/editors/distill_sd/distill_sd_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/distill_sd/distill_sd_xl.py -------------------------------------------------------------------------------- /diffengine/models/editors/esd/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/esd/__init__.py -------------------------------------------------------------------------------- /diffengine/models/editors/esd/esd_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/esd/esd_xl.py -------------------------------------------------------------------------------- /diffengine/models/editors/esd/esd_xl_data_preprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/esd/esd_xl_data_preprocessor.py -------------------------------------------------------------------------------- /diffengine/models/editors/instruct_pix2pix/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/instruct_pix2pix/__init__.py -------------------------------------------------------------------------------- /diffengine/models/editors/instruct_pix2pix/instruct_pix2pix_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/instruct_pix2pix/instruct_pix2pix_xl.py -------------------------------------------------------------------------------- /diffengine/models/editors/ip_adapter/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/ip_adapter/__init__.py -------------------------------------------------------------------------------- /diffengine/models/editors/ip_adapter/ip_adapter_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/ip_adapter/ip_adapter_xl.py -------------------------------------------------------------------------------- /diffengine/models/editors/ip_adapter/ip_adapter_xl_data_preprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/ip_adapter/ip_adapter_xl_data_preprocessor.py -------------------------------------------------------------------------------- /diffengine/models/editors/ip_adapter/ip_adapter_xl_timm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/ip_adapter/ip_adapter_xl_timm.py -------------------------------------------------------------------------------- /diffengine/models/editors/ip_adapter/pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/ip_adapter/pipeline.py -------------------------------------------------------------------------------- /diffengine/models/editors/kandinsky/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/kandinsky/__init__.py -------------------------------------------------------------------------------- /diffengine/models/editors/kandinsky/kandinskyv22_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/kandinsky/kandinskyv22_decoder.py -------------------------------------------------------------------------------- /diffengine/models/editors/kandinsky/kandinskyv22_decoder_preprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/kandinsky/kandinskyv22_decoder_preprocessor.py -------------------------------------------------------------------------------- /diffengine/models/editors/kandinsky/kandinskyv22_prior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/kandinsky/kandinskyv22_prior.py -------------------------------------------------------------------------------- /diffengine/models/editors/kandinsky/kandinskyv3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/kandinsky/kandinskyv3.py -------------------------------------------------------------------------------- /diffengine/models/editors/lcm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/lcm/__init__.py -------------------------------------------------------------------------------- /diffengine/models/editors/lcm/lcm_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/lcm/lcm_modules.py -------------------------------------------------------------------------------- /diffengine/models/editors/lcm/lcm_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/lcm/lcm_xl.py -------------------------------------------------------------------------------- /diffengine/models/editors/pixart_alpha/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/pixart_alpha/__init__.py -------------------------------------------------------------------------------- /diffengine/models/editors/pixart_alpha/pixart_alpha.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/pixart_alpha/pixart_alpha.py -------------------------------------------------------------------------------- /diffengine/models/editors/pixart_alpha/pixart_alpha_data_preprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/pixart_alpha/pixart_alpha_data_preprocessor.py -------------------------------------------------------------------------------- /diffengine/models/editors/ssd_1b/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/ssd_1b/__init__.py -------------------------------------------------------------------------------- /diffengine/models/editors/ssd_1b/ssd_1b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/ssd_1b/ssd_1b.py -------------------------------------------------------------------------------- /diffengine/models/editors/stable_diffusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/stable_diffusion/__init__.py -------------------------------------------------------------------------------- /diffengine/models/editors/stable_diffusion/sd_data_preprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/stable_diffusion/sd_data_preprocessor.py -------------------------------------------------------------------------------- /diffengine/models/editors/stable_diffusion/stable_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/stable_diffusion/stable_diffusion.py -------------------------------------------------------------------------------- /diffengine/models/editors/stable_diffusion_controlnet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/stable_diffusion_controlnet/__init__.py -------------------------------------------------------------------------------- /diffengine/models/editors/stable_diffusion_controlnet/sd_controlnet_data_preprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/stable_diffusion_controlnet/sd_controlnet_data_preprocessor.py -------------------------------------------------------------------------------- /diffengine/models/editors/stable_diffusion_controlnet/stable_diffusion_controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/stable_diffusion_controlnet/stable_diffusion_controlnet.py -------------------------------------------------------------------------------- /diffengine/models/editors/stable_diffusion_inpaint/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/stable_diffusion_inpaint/__init__.py -------------------------------------------------------------------------------- /diffengine/models/editors/stable_diffusion_inpaint/sd_inpaint_data_preprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/stable_diffusion_inpaint/sd_inpaint_data_preprocessor.py -------------------------------------------------------------------------------- /diffengine/models/editors/stable_diffusion_inpaint/stable_diffusion_inpaint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/stable_diffusion_inpaint/stable_diffusion_inpaint.py -------------------------------------------------------------------------------- /diffengine/models/editors/stable_diffusion_xl/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/stable_diffusion_xl/__init__.py -------------------------------------------------------------------------------- /diffengine/models/editors/stable_diffusion_xl/sdxl_data_preprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/stable_diffusion_xl/sdxl_data_preprocessor.py -------------------------------------------------------------------------------- /diffengine/models/editors/stable_diffusion_xl/stable_diffusion_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/stable_diffusion_xl/stable_diffusion_xl.py -------------------------------------------------------------------------------- /diffengine/models/editors/stable_diffusion_xl_controlnet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/stable_diffusion_xl_controlnet/__init__.py -------------------------------------------------------------------------------- /diffengine/models/editors/stable_diffusion_xl_controlnet/sdxl_controlnet_data_preprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/stable_diffusion_xl_controlnet/sdxl_controlnet_data_preprocessor.py -------------------------------------------------------------------------------- /diffengine/models/editors/stable_diffusion_xl_controlnet/stable_diffusion_xl_controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/stable_diffusion_xl_controlnet/stable_diffusion_xl_controlnet.py -------------------------------------------------------------------------------- /diffengine/models/editors/stable_diffusion_xl_dpo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/stable_diffusion_xl_dpo/__init__.py -------------------------------------------------------------------------------- /diffengine/models/editors/stable_diffusion_xl_dpo/sdxl_dpo_data_preprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/stable_diffusion_xl_dpo/sdxl_dpo_data_preprocessor.py -------------------------------------------------------------------------------- /diffengine/models/editors/stable_diffusion_xl_dpo/stable_diffusion_xl_dpo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/stable_diffusion_xl_dpo/stable_diffusion_xl_dpo.py -------------------------------------------------------------------------------- /diffengine/models/editors/stable_diffusion_xl_inpaint/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/stable_diffusion_xl_inpaint/__init__.py -------------------------------------------------------------------------------- /diffengine/models/editors/stable_diffusion_xl_inpaint/sdxl_inpaint_data_preprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/stable_diffusion_xl_inpaint/sdxl_inpaint_data_preprocessor.py -------------------------------------------------------------------------------- /diffengine/models/editors/stable_diffusion_xl_inpaint/stable_diffusion_xl_inpaint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/stable_diffusion_xl_inpaint/stable_diffusion_xl_inpaint.py -------------------------------------------------------------------------------- /diffengine/models/editors/t2i_adapter/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/t2i_adapter/__init__.py -------------------------------------------------------------------------------- /diffengine/models/editors/t2i_adapter/stable_diffusion_xl_t2i_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/t2i_adapter/stable_diffusion_xl_t2i_adapter.py -------------------------------------------------------------------------------- /diffengine/models/editors/wuerstchen/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/wuerstchen/__init__.py -------------------------------------------------------------------------------- /diffengine/models/editors/wuerstchen/efficient_net_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/wuerstchen/efficient_net_encoder.py -------------------------------------------------------------------------------- /diffengine/models/editors/wuerstchen/wuerstchen_prior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/editors/wuerstchen/wuerstchen_prior.py -------------------------------------------------------------------------------- /diffengine/models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/losses/__init__.py -------------------------------------------------------------------------------- /diffengine/models/losses/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/losses/base.py -------------------------------------------------------------------------------- /diffengine/models/losses/cross_entropy_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/losses/cross_entropy_loss.py -------------------------------------------------------------------------------- /diffengine/models/losses/debias_estimation_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/losses/debias_estimation_loss.py -------------------------------------------------------------------------------- /diffengine/models/losses/hubar_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/losses/hubar_loss.py -------------------------------------------------------------------------------- /diffengine/models/losses/l2_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/losses/l2_loss.py -------------------------------------------------------------------------------- /diffengine/models/losses/snr_l2_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/losses/snr_l2_loss.py -------------------------------------------------------------------------------- /diffengine/models/losses/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/losses/utils.py -------------------------------------------------------------------------------- /diffengine/models/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/utils/__init__.py -------------------------------------------------------------------------------- /diffengine/models/utils/noise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/utils/noise.py -------------------------------------------------------------------------------- /diffengine/models/utils/timesteps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/models/utils/timesteps.py -------------------------------------------------------------------------------- /diffengine/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/registry.py -------------------------------------------------------------------------------- /diffengine/tools/analysis_tools/show_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/tools/analysis_tools/show_vae.py -------------------------------------------------------------------------------- /diffengine/tools/copy_cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/tools/copy_cfg.py -------------------------------------------------------------------------------- /diffengine/tools/list_cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/tools/list_cfg.py -------------------------------------------------------------------------------- /diffengine/tools/model_converters/publish_model2diffusers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/tools/model_converters/publish_model2diffusers.py -------------------------------------------------------------------------------- /diffengine/tools/preprocess/bucket_ids.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/tools/preprocess/bucket_ids.py -------------------------------------------------------------------------------- /diffengine/tools/preprocess/csv_to_txt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/tools/preprocess/csv_to_txt.py -------------------------------------------------------------------------------- /diffengine/tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/tools/train.py -------------------------------------------------------------------------------- /diffengine/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/diffengine/version.py -------------------------------------------------------------------------------- /docker-compose.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docker-compose.yml -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/source/_templates/404.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/_templates/404.html -------------------------------------------------------------------------------- /docs/source/_templates/python/attribute.rst: -------------------------------------------------------------------------------- 1 | {% extends "python/data.rst" %} 2 | -------------------------------------------------------------------------------- /docs/source/_templates/python/class.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/_templates/python/class.rst -------------------------------------------------------------------------------- /docs/source/_templates/python/data.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/_templates/python/data.rst -------------------------------------------------------------------------------- /docs/source/_templates/python/exception.rst: -------------------------------------------------------------------------------- 1 | {% extends "python/class.rst" %} 2 | -------------------------------------------------------------------------------- /docs/source/_templates/python/function.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/_templates/python/function.rst -------------------------------------------------------------------------------- /docs/source/_templates/python/method.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/_templates/python/method.rst -------------------------------------------------------------------------------- /docs/source/_templates/python/module.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/_templates/python/module.rst -------------------------------------------------------------------------------- /docs/source/_templates/python/package.rst: -------------------------------------------------------------------------------- 1 | {% extends "python/module.rst" %} 2 | -------------------------------------------------------------------------------- /docs/source/_templates/python/property.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/_templates/python/property.rst -------------------------------------------------------------------------------- /docs/source/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/conf.py -------------------------------------------------------------------------------- /docs/source/get_started.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/get_started.md -------------------------------------------------------------------------------- /docs/source/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/index.rst -------------------------------------------------------------------------------- /docs/source/run_guides/inference.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/run_guides/inference.md -------------------------------------------------------------------------------- /docs/source/run_guides/run_controlnet.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/run_guides/run_controlnet.md -------------------------------------------------------------------------------- /docs/source/run_guides/run_controlnet_xl.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/run_guides/run_controlnet_xl.md -------------------------------------------------------------------------------- /docs/source/run_guides/run_dreambooth.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/run_guides/run_dreambooth.md -------------------------------------------------------------------------------- /docs/source/run_guides/run_dreambooth_xl.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/run_guides/run_dreambooth_xl.md -------------------------------------------------------------------------------- /docs/source/run_guides/run_instruct_pix2pix.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/run_guides/run_instruct_pix2pix.md -------------------------------------------------------------------------------- /docs/source/run_guides/run_ip_adapter.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/run_guides/run_ip_adapter.md -------------------------------------------------------------------------------- /docs/source/run_guides/run_kandinsky_v22.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/run_guides/run_kandinsky_v22.md -------------------------------------------------------------------------------- /docs/source/run_guides/run_lcm.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/run_guides/run_lcm.md -------------------------------------------------------------------------------- /docs/source/run_guides/run_lcm_lora.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/run_guides/run_lcm_lora.md -------------------------------------------------------------------------------- /docs/source/run_guides/run_lora.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/run_guides/run_lora.md -------------------------------------------------------------------------------- /docs/source/run_guides/run_lora_xl.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/run_guides/run_lora_xl.md -------------------------------------------------------------------------------- /docs/source/run_guides/run_pixart_alpha.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/run_guides/run_pixart_alpha.md -------------------------------------------------------------------------------- /docs/source/run_guides/run_pixart_alpha_dreambooth.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/run_guides/run_pixart_alpha_dreambooth.md -------------------------------------------------------------------------------- /docs/source/run_guides/run_pixart_alpha_lora.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/run_guides/run_pixart_alpha_lora.md -------------------------------------------------------------------------------- /docs/source/run_guides/run_sd.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/run_guides/run_sd.md -------------------------------------------------------------------------------- /docs/source/run_guides/run_sdxl.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/run_guides/run_sdxl.md -------------------------------------------------------------------------------- /docs/source/run_guides/run_t2i_adapter.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/run_guides/run_t2i_adapter.md -------------------------------------------------------------------------------- /docs/source/run_guides/run_wuerstchen.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/run_guides/run_wuerstchen.md -------------------------------------------------------------------------------- /docs/source/run_guides/run_wuerstchen_lora.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/run_guides/run_wuerstchen_lora.md -------------------------------------------------------------------------------- /docs/source/user_guides/aspect_ratio_bucketing.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/user_guides/aspect_ratio_bucketing.md -------------------------------------------------------------------------------- /docs/source/user_guides/config.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/user_guides/config.md -------------------------------------------------------------------------------- /docs/source/user_guides/dataset_prepare.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/docs/source/user_guides/dataset_prepare.md -------------------------------------------------------------------------------- /projects/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/projects/README.md -------------------------------------------------------------------------------- /projects/controlnetxs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/projects/controlnetxs/README.md -------------------------------------------------------------------------------- /projects/controlnetxs/_base_/stable_diffusion_xl_controlnetxs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/projects/controlnetxs/_base_/stable_diffusion_xl_controlnetxs.py -------------------------------------------------------------------------------- /projects/controlnetxs/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/projects/controlnetxs/modules/__init__.py -------------------------------------------------------------------------------- /projects/controlnetxs/modules/controlnetxs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/projects/controlnetxs/modules/controlnetxs.py -------------------------------------------------------------------------------- /projects/controlnetxs/modules/pipeline_controlnet_xs_sd_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/projects/controlnetxs/modules/pipeline_controlnet_xs_sd_xl.py -------------------------------------------------------------------------------- /projects/controlnetxs/modules/stable_diffusion_xl_controlnetxs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/projects/controlnetxs/modules/stable_diffusion_xl_controlnetxs.py -------------------------------------------------------------------------------- /projects/controlnetxs/stable_diffusion_xl_controlnetxs_fill50k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/projects/controlnetxs/stable_diffusion_xl_controlnetxs_fill50k.py -------------------------------------------------------------------------------- /projects/face_expression/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/projects/face_expression/README.md -------------------------------------------------------------------------------- /projects/face_expression/_base_/anythingv5_lora_textencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/projects/face_expression/_base_/anythingv5_lora_textencoder.py -------------------------------------------------------------------------------- /projects/face_expression/_base_/counterfeit_xl_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/projects/face_expression/_base_/counterfeit_xl_lora.py -------------------------------------------------------------------------------- /projects/face_expression/_base_/face_expression_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/projects/face_expression/_base_/face_expression_dataset.py -------------------------------------------------------------------------------- /projects/face_expression/_base_/face_expression_xl_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/projects/face_expression/_base_/face_expression_xl_dataset.py -------------------------------------------------------------------------------- /projects/face_expression/stable_diffusion_v15_lora_face_expression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/projects/face_expression/stable_diffusion_v15_lora_face_expression.py -------------------------------------------------------------------------------- /projects/face_expression/stable_diffusion_xl_lora_face_expression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/projects/face_expression/stable_diffusion_xl_lora_face_expression.py -------------------------------------------------------------------------------- /projects/tohoku_zunko_project/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/projects/tohoku_zunko_project/README.md -------------------------------------------------------------------------------- /projects/tohoku_zunko_project/_base_/anythingv5_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/projects/tohoku_zunko_project/_base_/anythingv5_lora.py -------------------------------------------------------------------------------- /projects/tohoku_zunko_project/_base_/counterfeit_xl_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/projects/tohoku_zunko_project/_base_/counterfeit_xl_lora.py -------------------------------------------------------------------------------- /projects/tohoku_zunko_project/_base_/zunko_dreambooth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/projects/tohoku_zunko_project/_base_/zunko_dreambooth.py -------------------------------------------------------------------------------- /projects/tohoku_zunko_project/_base_/zunko_dreambooth_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/projects/tohoku_zunko_project/_base_/zunko_dreambooth_xl.py -------------------------------------------------------------------------------- /projects/tohoku_zunko_project/anythingv5_dreambooth_lora_zunko.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/projects/tohoku_zunko_project/anythingv5_dreambooth_lora_zunko.py -------------------------------------------------------------------------------- /projects/tohoku_zunko_project/counterfeit_xl_dreambooth_lora_zunko.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/projects/tohoku_zunko_project/counterfeit_xl_dreambooth_lora_zunko.py -------------------------------------------------------------------------------- /projects/tohoku_zunko_project/stable_diffusion_v15_dreambooth_lora_zunko.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/projects/tohoku_zunko_project/stable_diffusion_v15_dreambooth_lora_zunko.py -------------------------------------------------------------------------------- /projects/tohoku_zunko_project/stable_diffusion_xl_dreambooth_lora_zunko.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/projects/tohoku_zunko_project/stable_diffusion_xl_dreambooth_lora_zunko.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/pyproject.toml -------------------------------------------------------------------------------- /tests/configs/ip_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/configs/ip_adapter.py -------------------------------------------------------------------------------- /tests/configs/pixart.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/configs/pixart.py -------------------------------------------------------------------------------- /tests/configs/sd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/configs/sd.py -------------------------------------------------------------------------------- /tests/configs/sdcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/configs/sdcn.py -------------------------------------------------------------------------------- /tests/configs/sdxl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/configs/sdxl.py -------------------------------------------------------------------------------- /tests/configs/sdxl_t2iadapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/configs/sdxl_t2iadapter.py -------------------------------------------------------------------------------- /tests/configs/sdxlcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/configs/sdxlcn.py -------------------------------------------------------------------------------- /tests/test_datasets/test_hf_controlnet_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_datasets/test_hf_controlnet_datasets.py -------------------------------------------------------------------------------- /tests/test_datasets/test_hf_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_datasets/test_hf_datasets.py -------------------------------------------------------------------------------- /tests/test_datasets/test_hf_dreambooth_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_datasets/test_hf_dreambooth_datasets.py -------------------------------------------------------------------------------- /tests/test_datasets/test_hf_esd_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_datasets/test_hf_esd_datasets.py -------------------------------------------------------------------------------- /tests/test_datasets/test_samplers/test_batch_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_datasets/test_samplers/test_batch_sampler.py -------------------------------------------------------------------------------- /tests/test_datasets/test_transforms/test_formatting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_datasets/test_transforms/test_formatting.py -------------------------------------------------------------------------------- /tests/test_datasets/test_transforms/test_loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_datasets/test_transforms/test_loading.py -------------------------------------------------------------------------------- /tests/test_datasets/test_transforms/test_processing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_datasets/test_transforms/test_processing.py -------------------------------------------------------------------------------- /tests/test_datasets/test_transforms/test_wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_datasets/test_transforms/test_wrappers.py -------------------------------------------------------------------------------- /tests/test_engine/test_hooks/test_compile_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_engine/test_hooks/test_compile_hook.py -------------------------------------------------------------------------------- /tests/test_engine/test_hooks/test_controlnet_save_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_engine/test_hooks/test_controlnet_save_hook.py -------------------------------------------------------------------------------- /tests/test_engine/test_hooks/test_fast_norm_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_engine/test_hooks/test_fast_norm_hook.py -------------------------------------------------------------------------------- /tests/test_engine/test_hooks/test_ip_adapter_save_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_engine/test_hooks/test_ip_adapter_save_hook.py -------------------------------------------------------------------------------- /tests/test_engine/test_hooks/test_lcm_ema_update_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_engine/test_hooks/test_lcm_ema_update_hook.py -------------------------------------------------------------------------------- /tests/test_engine/test_hooks/test_peft_save_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_engine/test_hooks/test_peft_save_hook.py -------------------------------------------------------------------------------- /tests/test_engine/test_hooks/test_sd_checkpoint_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_engine/test_hooks/test_sd_checkpoint_hook.py -------------------------------------------------------------------------------- /tests/test_engine/test_hooks/test_t2i_adapter_save_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_engine/test_hooks/test_t2i_adapter_save_hook.py -------------------------------------------------------------------------------- /tests/test_engine/test_hooks/test_transformer_checkpoint_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_engine/test_hooks/test_transformer_checkpoint_hook.py -------------------------------------------------------------------------------- /tests/test_engine/test_hooks/test_unet_ema_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_engine/test_hooks/test_unet_ema_hook.py -------------------------------------------------------------------------------- /tests/test_engine/test_hooks/test_visualization_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_engine/test_hooks/test_visualization_hook.py -------------------------------------------------------------------------------- /tests/test_engine/test_hooks/test_wuerstchen_save_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_engine/test_hooks/test_wuerstchen_save_hook.py -------------------------------------------------------------------------------- /tests/test_engine/test_optimizers/test_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_engine/test_optimizers/test_optimizer.py -------------------------------------------------------------------------------- /tests/test_models/test_archs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_models/test_archs.py -------------------------------------------------------------------------------- /tests/test_models/test_editors/test_amused/test_amused.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_models/test_editors/test_amused/test_amused.py -------------------------------------------------------------------------------- /tests/test_models/test_editors/test_deepfloyd_if/test_deepfloyd_if.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_models/test_editors/test_deepfloyd_if/test_deepfloyd_if.py -------------------------------------------------------------------------------- /tests/test_models/test_editors/test_distill_sd/test_distill_sd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_models/test_editors/test_distill_sd/test_distill_sd.py -------------------------------------------------------------------------------- /tests/test_models/test_editors/test_esd/test_esd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_models/test_editors/test_esd/test_esd.py -------------------------------------------------------------------------------- /tests/test_models/test_editors/test_instruct_pix2pix/test_instruct_pix2pix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_models/test_editors/test_instruct_pix2pix/test_instruct_pix2pix.py -------------------------------------------------------------------------------- /tests/test_models/test_editors/test_ip_adapter/test_ip_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_models/test_editors/test_ip_adapter/test_ip_adapter.py -------------------------------------------------------------------------------- /tests/test_models/test_editors/test_ip_adapter/test_ip_adapter_plus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_models/test_editors/test_ip_adapter/test_ip_adapter_plus.py -------------------------------------------------------------------------------- /tests/test_models/test_editors/test_ip_adapter/test_ip_adapter_plus_timm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_models/test_editors/test_ip_adapter/test_ip_adapter_plus_timm.py -------------------------------------------------------------------------------- /tests/test_models/test_editors/test_kandinsky/test_kandinsky_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_models/test_editors/test_kandinsky/test_kandinsky_v3.py -------------------------------------------------------------------------------- /tests/test_models/test_editors/test_kandinsky/test_kandinskyv22_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_models/test_editors/test_kandinsky/test_kandinskyv22_decoder.py -------------------------------------------------------------------------------- /tests/test_models/test_editors/test_kandinsky/test_kandinskyv22_prior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_models/test_editors/test_kandinsky/test_kandinskyv22_prior.py -------------------------------------------------------------------------------- /tests/test_models/test_editors/test_lcm/test_lcm_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_models/test_editors/test_lcm/test_lcm_xl.py -------------------------------------------------------------------------------- /tests/test_models/test_editors/test_pixart_alpha/test_pixart_alpha.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_models/test_editors/test_pixart_alpha/test_pixart_alpha.py -------------------------------------------------------------------------------- /tests/test_models/test_editors/test_ssd_1b/test_ssd_1b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_models/test_editors/test_ssd_1b/test_ssd_1b.py -------------------------------------------------------------------------------- /tests/test_models/test_editors/test_stable_diffusion/test_stable_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_models/test_editors/test_stable_diffusion/test_stable_diffusion.py -------------------------------------------------------------------------------- /tests/test_models/test_editors/test_stable_diffusion_controlnet/test_stable_diffusion_controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_models/test_editors/test_stable_diffusion_controlnet/test_stable_diffusion_controlnet.py -------------------------------------------------------------------------------- /tests/test_models/test_editors/test_stable_diffusion_inpaint/test_stable_diffusion_inpaint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_models/test_editors/test_stable_diffusion_inpaint/test_stable_diffusion_inpaint.py -------------------------------------------------------------------------------- /tests/test_models/test_editors/test_stable_diffusion_xl/test_stable_diffusion_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_models/test_editors/test_stable_diffusion_xl/test_stable_diffusion_xl.py -------------------------------------------------------------------------------- /tests/test_models/test_editors/test_stable_diffusion_xl_controlnet/test_stable_diffusion_xl_controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_models/test_editors/test_stable_diffusion_xl_controlnet/test_stable_diffusion_xl_controlnet.py -------------------------------------------------------------------------------- /tests/test_models/test_editors/test_stable_diffusion_xl_dpo/test_stable_diffusion_xl_dpo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_models/test_editors/test_stable_diffusion_xl_dpo/test_stable_diffusion_xl_dpo.py -------------------------------------------------------------------------------- /tests/test_models/test_editors/test_stable_diffusion_xl_inpaint/test_stable_diffusion_xl_inpaint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_models/test_editors/test_stable_diffusion_xl_inpaint/test_stable_diffusion_xl_inpaint.py -------------------------------------------------------------------------------- /tests/test_models/test_editors/test_t2i_adapter/test_stable_diffusion_xl_t2i_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_models/test_editors/test_t2i_adapter/test_stable_diffusion_xl_t2i_adapter.py -------------------------------------------------------------------------------- /tests/test_models/test_editors/test_wuerstchen/test_wuerstchen_prior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_models/test_editors/test_wuerstchen/test_wuerstchen_prior.py -------------------------------------------------------------------------------- /tests/test_models/test_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_models/test_losses.py -------------------------------------------------------------------------------- /tests/test_models/test_utils/test_noise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_models/test_utils/test_noise.py -------------------------------------------------------------------------------- /tests/test_models/test_utils/test_timesteps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/test_models/test_utils/test_timesteps.py -------------------------------------------------------------------------------- /tests/testdata/bucket_ids.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/testdata/bucket_ids.pkl -------------------------------------------------------------------------------- /tests/testdata/color.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/testdata/color.jpg -------------------------------------------------------------------------------- /tests/testdata/cond.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/testdata/cond.jpg -------------------------------------------------------------------------------- /tests/testdata/dataset/color.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/testdata/dataset/color.jpg -------------------------------------------------------------------------------- /tests/testdata/dataset/metadata.csv: -------------------------------------------------------------------------------- 1 | file_name,text 2 | color.jpg,"a dog" 3 | -------------------------------------------------------------------------------- /tests/testdata/dataset/metadata2.csv: -------------------------------------------------------------------------------- 1 | file_name,text 2 | color.jpg,"a cat" 3 | -------------------------------------------------------------------------------- /tests/testdata/dataset/metadata_cn.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/testdata/dataset/metadata_cn.csv -------------------------------------------------------------------------------- /tests/testdata/dataset_db/color.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/okotaku/diffengine/HEAD/tests/testdata/dataset_db/color.jpg --------------------------------------------------------------------------------