where is the code for local attention loss?

ZYM-PKU / UDiffText

UDiffText: A Unified Framework for High-quality Text Synthesis in Arbitrary Images via Character-aware Diffusion Models

MIT License

192 stars 16 forks source link

where is the code for local attention loss? #4

Closed ZhendongWang6 closed 8 months ago

ZhendongWang6 commented 8 months ago

Hi, it is a great work! But I do not find the implementation of the local attention loss. Could you help me? Thanks.

ZYM-PKU commented 8 months ago

Much appreciated. The implementation of the local attention loss can be found in ./sgm/modules/diffusionmodules/loss.py. Check the function get_local_loss().