where is attention_contacts in esm2?

[linshuxi]

Hi, thank you for the great job! There is a return which contains attention_contacts in esm2, and the code is: results = model(batch_tokens, repr_layers=[33], return_contacts=True). However, i cannot find similar thing in esm3. Could you give an example on how to get similar results of esm2 in esm3?