Coding Stable Diffusion from scratch in PyTorch