Try to modify the attention module in the method to my own innovative SSAM-like module, what files need to be modified?

Jamie-Cheung / ArtBank

ArtBank: Artistic Style Transfer with Pre-trained Diffusion Model and Implicit Style Prompt Bank (AAAI2024)

27 stars 2 forks source link

Try to modify the attention module in the method to my own innovative SSAM-like module, what files need to be modified? #7

Open BBABM opened 2 weeks ago

BBABM commented 2 weeks ago

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!

Jamie-Cheung commented 1 week ago

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!

Yes. Only modify an attention module and define a learnable parameter.

BBABM commented 1 week ago

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!

Yes. Only modify an attention module and define a learnable parameter.

thank you!