run mvdream/configs/sd-v2-base.yaml, it occurs attn_mask does not match input{ (77, 77) and (77, 16, 1, 1) }

The problem occurs in mvdream/ldm/encoders/modules.py:

run x = r(x, attn_mask=attn_mask) this line, it will occur not match!!!
when I modify r(x, attn_mask=attn_mask) to r(x, attn_mask=None), it's okay. But will it affect the final result?