---
license: mit
---
# LEGO-Diffusion checkpoint repository

### For implementation details, please refer to our [Github Repo](https://github.com/JegZheng/LEGODiffusion)


#### ImageNet pretrained checkpoints
<table><tbody>
<!-- START TABLE -->
<!-- TABLE HEADER -->
<th align="left">model</th>
<th align="left">FID</th>
<th align="left">IS</th>
<!-- TABLE BODY -->
<tr>
<td align="left">LEGO-L-PG</td>
<td align="left">2.16</td>
<td align="left">78.7</td>
</tr>
<tr>
<td align="left">LEGO-L-PR</td>
<td align="left">2.29</td>
<td align="left">78.7</td>
</tr>
<tr>
<td align="left">LEGO-XL-U</td>
<td align="left">2.59</td>
<td align="left">338.08</td>
</tr>
</tbody></table>


### BibTex
Please cite our work if you find it is helpful. Thank you!
```
@inproceedings{
zheng2024learning,
title={Learning Stackable and Skippable {LEGO} Bricks for Efficient, Reconfigurable, and Variable-Resolution Diffusion Modeling},
author={Huangjie Zheng and Zhendong Wang and Jianbo Yuan and Guanghan Ning and Pengcheng He and Quanzeng You and Hongxia Yang and Mingyuan Zhou},
booktitle={The Twelfth International Conference on Learning Representations},
year={2024},
url={https://openreview.net/forum?id=qmXedvwrT1}
}
```