├── .gitignore ├── LICENSE.txt ├── README.md ├── assets ├── idea.jpg ├── quality.jpg ├── speedups.jpg └── teaser.jpg ├── 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 ├── pipelines.py └── utils.py ├── scripts ├── compute_metrics.py ├── dump_coco.py ├── export_html.py ├── generate_coco.py ├── profile_macs.py ├── run_sdxl.py ├── sd_example.py └── sdxl_example.py └── setup.py /.gitignore: -------------------------------------------------------------------------------- 1 | .idea 2 | .DS_Store 3 | build 4 | dist 5 | *.egg-info -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/distrifuser/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/distrifuser/HEAD/README.md -------------------------------------------------------------------------------- /assets/idea.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/distrifuser/HEAD/assets/idea.jpg -------------------------------------------------------------------------------- /assets/quality.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/distrifuser/HEAD/assets/quality.jpg -------------------------------------------------------------------------------- /assets/speedups.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/distrifuser/HEAD/assets/speedups.jpg -------------------------------------------------------------------------------- /assets/teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/distrifuser/HEAD/assets/teaser.jpg -------------------------------------------------------------------------------- /distrifuser/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /distrifuser/__version__.py: -------------------------------------------------------------------------------- 1 | __version__ = "0.0.1beta1" 2 | -------------------------------------------------------------------------------- /distrifuser/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /distrifuser/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/distrifuser/HEAD/distrifuser/models/base_model.py -------------------------------------------------------------------------------- /distrifuser/models/distri_sdxl_unet_pp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/distrifuser/HEAD/distrifuser/models/distri_sdxl_unet_pp.py -------------------------------------------------------------------------------- /distrifuser/models/distri_sdxl_unet_tp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/distrifuser/HEAD/distrifuser/models/distri_sdxl_unet_tp.py -------------------------------------------------------------------------------- /distrifuser/models/naive_patch_sdxl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/distrifuser/HEAD/distrifuser/models/naive_patch_sdxl.py -------------------------------------------------------------------------------- /distrifuser/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /distrifuser/modules/base_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/distrifuser/HEAD/distrifuser/modules/base_module.py -------------------------------------------------------------------------------- /distrifuser/modules/pp/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /distrifuser/modules/pp/attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/distrifuser/HEAD/distrifuser/modules/pp/attn.py -------------------------------------------------------------------------------- /distrifuser/modules/pp/conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/distrifuser/HEAD/distrifuser/modules/pp/conv2d.py -------------------------------------------------------------------------------- /distrifuser/modules/pp/groupnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/distrifuser/HEAD/distrifuser/modules/pp/groupnorm.py -------------------------------------------------------------------------------- /distrifuser/modules/tp/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /distrifuser/modules/tp/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/distrifuser/HEAD/distrifuser/modules/tp/attention.py -------------------------------------------------------------------------------- /distrifuser/modules/tp/conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/distrifuser/HEAD/distrifuser/modules/tp/conv2d.py -------------------------------------------------------------------------------- /distrifuser/modules/tp/feed_forward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/distrifuser/HEAD/distrifuser/modules/tp/feed_forward.py -------------------------------------------------------------------------------- /distrifuser/modules/tp/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/distrifuser/HEAD/distrifuser/modules/tp/resnet.py -------------------------------------------------------------------------------- /distrifuser/pipelines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/distrifuser/HEAD/distrifuser/pipelines.py -------------------------------------------------------------------------------- /distrifuser/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/distrifuser/HEAD/distrifuser/utils.py -------------------------------------------------------------------------------- /scripts/compute_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/distrifuser/HEAD/scripts/compute_metrics.py -------------------------------------------------------------------------------- /scripts/dump_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/distrifuser/HEAD/scripts/dump_coco.py -------------------------------------------------------------------------------- /scripts/export_html.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/distrifuser/HEAD/scripts/export_html.py -------------------------------------------------------------------------------- /scripts/generate_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/distrifuser/HEAD/scripts/generate_coco.py -------------------------------------------------------------------------------- /scripts/profile_macs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/distrifuser/HEAD/scripts/profile_macs.py -------------------------------------------------------------------------------- /scripts/run_sdxl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/distrifuser/HEAD/scripts/run_sdxl.py -------------------------------------------------------------------------------- /scripts/sd_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/distrifuser/HEAD/scripts/sd_example.py -------------------------------------------------------------------------------- /scripts/sdxl_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/distrifuser/HEAD/scripts/sdxl_example.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/distrifuser/HEAD/setup.py --------------------------------------------------------------------------------