consult SAE lens / other implementations we will need a diffusion model decoder at the end
consult SAE lens / other implementations
we will need a diffusion model decoder at the end