extend_for_diffusion.py 2.21 KB