Do any of you know which files I should change which places if I want to replace the cross attention module in this method

Is it just a matter of modifying the ldm/modules/attention.py，ldm/modules/embedding_manager.py   ldm/modules/encoders/modules.py， ldm/models/diffusion/ddpm.py . Help is desperately needed, thanks!