feat(pt): type embedding can still be compress even if attn_layer != 0#5066

wanghan-iapcm merged 4 commits intodeepmodeling:develdeepmodeling/deepmd-kit:develfrom

OutisLi:pr/atten_compOutisLi/deepmd-kit:pr/atten_compCopy head branch name to clipboard