diffusion models use trace molecule instead of atom, not running on start

[#47400721]