What are the limitations of using U-Nets in diffusion models, and how can other network architectures potentially improve performance