Fix crash when using UNetSelfAttentionMultiply (#14823)