├── LICENSE ├── README.md ├── assets ├── captions.txt ├── diffusers_linfusion_example.jpg ├── laion_improved_aesthetics_6.5plus_with_images_blip_captions.json └── picture.png ├── examples ├── eval │ ├── calculate_metrics.sh │ └── eval.sh ├── inference │ ├── basic_usage.ipynb │ ├── basic_usage_sd2.ipynb │ ├── basic_usage_sdxl.ipynb │ ├── sdxl_distrifusion_example.py │ ├── superres_sdxl.ipynb │ ├── superres_sdxl_low_w_mem.ipynb │ ├── ultra_text2image_sdxl.ipynb │ └── ultra_text2image_w_sdedit.ipynb └── train │ ├── distill.sh │ └── distill_sdxl.sh ├── requirements.txt └── src ├── distrifuser ├── __init__.py ├── __version__.py ├── models │ ├── __init__.py │ ├── base_model.py │ ├── distri_sdxl_unet_pp.py │ ├── distri_sdxl_unet_tp.py │ └── naive_patch_sdxl.py ├── modules │ ├── __init__.py │ ├── base_module.py │ ├── pp │ │ ├── __init__.py │ │ ├── attn.py │ │ ├── conv2d.py │ │ └── groupnorm.py │ └── tp │ │ ├── __init__.py │ │ ├── attention.py │ │ ├── conv2d.py │ │ ├── feed_forward.py │ │ └── resnet.py └── utils.py ├── eval ├── __init__.py ├── calculate_metrics.py ├── custom_ops.py ├── data_util.py ├── distributed.py ├── dnnlib │ ├── __init__.py │ └── util.py ├── eval.py ├── misc.py ├── ops │ ├── __init__.py │ ├── bias_act.cpp │ ├── bias_act.cu │ ├── bias_act.h │ ├── bias_act.py │ ├── conv2d_gradfix.py │ ├── conv2d_resample.py │ ├── fma.py │ ├── grid_sample_gradfix.py │ ├── upfirdn2d.cpp │ ├── upfirdn2d.cu │ ├── upfirdn2d.h │ └── upfirdn2d.py ├── persistence.py └── training_stats.py ├── linfusion ├── __init__.py ├── attention.py └── linfusion.py ├── pipelines ├── pipeline_highres_sdxl.py ├── pipeline_superres_sdxl.py └── pipelines_distrifusion_sdxl.py ├── tools.py └── train ├── distill.py └── distill_sdxl.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/README.md -------------------------------------------------------------------------------- /assets/captions.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/assets/captions.txt -------------------------------------------------------------------------------- /assets/diffusers_linfusion_example.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/assets/diffusers_linfusion_example.jpg -------------------------------------------------------------------------------- /assets/laion_improved_aesthetics_6.5plus_with_images_blip_captions.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/assets/laion_improved_aesthetics_6.5plus_with_images_blip_captions.json -------------------------------------------------------------------------------- /assets/picture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/assets/picture.png -------------------------------------------------------------------------------- /examples/eval/calculate_metrics.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/examples/eval/calculate_metrics.sh -------------------------------------------------------------------------------- /examples/eval/eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/examples/eval/eval.sh -------------------------------------------------------------------------------- /examples/inference/basic_usage.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/examples/inference/basic_usage.ipynb -------------------------------------------------------------------------------- /examples/inference/basic_usage_sd2.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/examples/inference/basic_usage_sd2.ipynb -------------------------------------------------------------------------------- /examples/inference/basic_usage_sdxl.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/examples/inference/basic_usage_sdxl.ipynb -------------------------------------------------------------------------------- /examples/inference/sdxl_distrifusion_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/examples/inference/sdxl_distrifusion_example.py -------------------------------------------------------------------------------- /examples/inference/superres_sdxl.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/examples/inference/superres_sdxl.ipynb -------------------------------------------------------------------------------- /examples/inference/superres_sdxl_low_w_mem.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/examples/inference/superres_sdxl_low_w_mem.ipynb -------------------------------------------------------------------------------- /examples/inference/ultra_text2image_sdxl.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/examples/inference/ultra_text2image_sdxl.ipynb -------------------------------------------------------------------------------- /examples/inference/ultra_text2image_w_sdedit.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/examples/inference/ultra_text2image_w_sdedit.ipynb -------------------------------------------------------------------------------- /examples/train/distill.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/examples/train/distill.sh -------------------------------------------------------------------------------- /examples/train/distill_sdxl.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/examples/train/distill_sdxl.sh -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/distrifuser/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/distrifuser/__version__.py: -------------------------------------------------------------------------------- 1 | __version__ = "0.0.1beta1" 2 | -------------------------------------------------------------------------------- /src/distrifuser/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/distrifuser/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/distrifuser/models/base_model.py -------------------------------------------------------------------------------- /src/distrifuser/models/distri_sdxl_unet_pp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/distrifuser/models/distri_sdxl_unet_pp.py -------------------------------------------------------------------------------- /src/distrifuser/models/distri_sdxl_unet_tp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/distrifuser/models/distri_sdxl_unet_tp.py -------------------------------------------------------------------------------- /src/distrifuser/models/naive_patch_sdxl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/distrifuser/models/naive_patch_sdxl.py -------------------------------------------------------------------------------- /src/distrifuser/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/distrifuser/modules/base_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/distrifuser/modules/base_module.py -------------------------------------------------------------------------------- /src/distrifuser/modules/pp/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/distrifuser/modules/pp/attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/distrifuser/modules/pp/attn.py -------------------------------------------------------------------------------- /src/distrifuser/modules/pp/conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/distrifuser/modules/pp/conv2d.py -------------------------------------------------------------------------------- /src/distrifuser/modules/pp/groupnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/distrifuser/modules/pp/groupnorm.py -------------------------------------------------------------------------------- /src/distrifuser/modules/tp/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/distrifuser/modules/tp/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/distrifuser/modules/tp/attention.py -------------------------------------------------------------------------------- /src/distrifuser/modules/tp/conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/distrifuser/modules/tp/conv2d.py -------------------------------------------------------------------------------- /src/distrifuser/modules/tp/feed_forward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/distrifuser/modules/tp/feed_forward.py -------------------------------------------------------------------------------- /src/distrifuser/modules/tp/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/distrifuser/modules/tp/resnet.py -------------------------------------------------------------------------------- /src/distrifuser/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/distrifuser/utils.py -------------------------------------------------------------------------------- /src/eval/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/eval/__init__.py -------------------------------------------------------------------------------- /src/eval/calculate_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/eval/calculate_metrics.py -------------------------------------------------------------------------------- /src/eval/custom_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/eval/custom_ops.py -------------------------------------------------------------------------------- /src/eval/data_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/eval/data_util.py -------------------------------------------------------------------------------- /src/eval/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/eval/distributed.py -------------------------------------------------------------------------------- /src/eval/dnnlib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/eval/dnnlib/__init__.py -------------------------------------------------------------------------------- /src/eval/dnnlib/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/eval/dnnlib/util.py -------------------------------------------------------------------------------- /src/eval/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/eval/eval.py -------------------------------------------------------------------------------- /src/eval/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/eval/misc.py -------------------------------------------------------------------------------- /src/eval/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/eval/ops/__init__.py -------------------------------------------------------------------------------- /src/eval/ops/bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/eval/ops/bias_act.cpp -------------------------------------------------------------------------------- /src/eval/ops/bias_act.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/eval/ops/bias_act.cu -------------------------------------------------------------------------------- /src/eval/ops/bias_act.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/eval/ops/bias_act.h -------------------------------------------------------------------------------- /src/eval/ops/bias_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/eval/ops/bias_act.py -------------------------------------------------------------------------------- /src/eval/ops/conv2d_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/eval/ops/conv2d_gradfix.py -------------------------------------------------------------------------------- /src/eval/ops/conv2d_resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/eval/ops/conv2d_resample.py -------------------------------------------------------------------------------- /src/eval/ops/fma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/eval/ops/fma.py -------------------------------------------------------------------------------- /src/eval/ops/grid_sample_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/eval/ops/grid_sample_gradfix.py -------------------------------------------------------------------------------- /src/eval/ops/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/eval/ops/upfirdn2d.cpp -------------------------------------------------------------------------------- /src/eval/ops/upfirdn2d.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/eval/ops/upfirdn2d.cu -------------------------------------------------------------------------------- /src/eval/ops/upfirdn2d.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/eval/ops/upfirdn2d.h -------------------------------------------------------------------------------- /src/eval/ops/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/eval/ops/upfirdn2d.py -------------------------------------------------------------------------------- /src/eval/persistence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/eval/persistence.py -------------------------------------------------------------------------------- /src/eval/training_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/eval/training_stats.py -------------------------------------------------------------------------------- /src/linfusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/linfusion/__init__.py -------------------------------------------------------------------------------- /src/linfusion/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/linfusion/attention.py -------------------------------------------------------------------------------- /src/linfusion/linfusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/linfusion/linfusion.py -------------------------------------------------------------------------------- /src/pipelines/pipeline_highres_sdxl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/pipelines/pipeline_highres_sdxl.py -------------------------------------------------------------------------------- /src/pipelines/pipeline_superres_sdxl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/pipelines/pipeline_superres_sdxl.py -------------------------------------------------------------------------------- /src/pipelines/pipelines_distrifusion_sdxl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/pipelines/pipelines_distrifusion_sdxl.py -------------------------------------------------------------------------------- /src/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/tools.py -------------------------------------------------------------------------------- /src/train/distill.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/train/distill.py -------------------------------------------------------------------------------- /src/train/distill_sdxl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Huage001/LinFusion/HEAD/src/train/distill_sdxl.py --------------------------------------------------------------------------------