feat: Enhance SceneEncoder with optional query prediction and integrate into Stage2Diffusion 89bdc42 jskvrna commited on May 25