Stable Diffusion Python 3 Math