WaterKnight/diffusion-models at main